hongshaorou/dreamer4
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

disable until rnn hiddens are handled properly

Browse Source
This commit is contained in:
lucidrains 2025-11-10 15:52:43 -08:00
parent 3c84b404a8
commit ac1c12f743
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
dreamer4/dreamer4.py
View File

@ -2229,6 +2229,7 @@ class DynamicsWorldModel(Module):
num_special_spatial_tokens = num_agents, num_special_spatial_tokens = num_agents,
time_block_every = time_block_every, time_block_every = time_block_every,
final_norm = False, final_norm = False,
rnn_time = False,
**transformer_kwargs **transformer_kwargs
) )

2
pyproject.toml
View File

@ -1,6 +1,6 @@
[project] [project]
name = "dreamer4" name = "dreamer4"
version = "0.1.17" version = "0.1.18"
description = "Dreamer 4" description = "Dreamer 4"
authors = [ authors = [
{ name = "Phil Wang", email = "lucidrains@gmail.com" } { name = "Phil Wang", email = "lucidrains@gmail.com" }