From 7195bbb1969070c8a2a414003f4c345c2c2f7166 Mon Sep 17 00:00:00 2001 From: lucidrains Date: Mon, 20 Oct 2025 12:42:27 -0700 Subject: [PATCH] oops --- dreamer4/dreamer4.py | 4 ++-- pyproject.toml | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/dreamer4/dreamer4.py b/dreamer4/dreamer4.py index 652aa02..d839f59 100644 --- a/dreamer4/dreamer4.py +++ b/dreamer4/dreamer4.py @@ -1949,7 +1949,7 @@ class DynamicsWorldModel(Module): discrete_actions, continuous_actions = actions - _, agent_embed = self.forward( + _, (agent_embed, _) = self.forward( latents = latents, signal_levels = self.max_steps - 1, step_sizes = step_size, @@ -1958,7 +1958,7 @@ class DynamicsWorldModel(Module): continuous_actions = continuous_actions, latent_is_noised = True, return_pred_only = True, - return_agent_tokens = True + return_intermediates = True ) agent_embed = agent_embed[..., agent_index, :] diff --git a/pyproject.toml b/pyproject.toml index 8dccab7..71e9c95 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [project] name = "dreamer4" -version = "0.0.49" +version = "0.0.50" description = "Dreamer 4" authors = [ { name = "Phil Wang", email = "lucidrains@gmail.com" }