disable until rnn hiddens are handled properly

2025-11-10 15:52:43 -08:00 · 2025-11-10 15:52:43 -08:00 · ac1c12f743
commit ac1c12f743
parent 3c84b404a8
2 changed files with 2 additions and 1 deletions
--- a/dreamer4/dreamer4.py
+++ b/dreamer4/dreamer4.py
@ -2229,6 +2229,7 @@ class DynamicsWorldModel(Module):
            num_special_spatial_tokens = num_agents,
            time_block_every = time_block_every,
            final_norm = False,
+            rnn_time = False,
            **transformer_kwargs
        )

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