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

oops

Browse Source
This commit is contained in:
lucidrains 2025-10-20 12:42:27 -07:00
parent ca244a290c
commit 7195bbb196
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
dreamer4/dreamer4.py
View File

@ -1949,7 +1949,7 @@ class DynamicsWorldModel(Module):
discrete_actions, continuous_actions = actions discrete_actions, continuous_actions = actions
_, agent_embed = self.forward( _, (agent_embed, _) = self.forward(
latents = latents, latents = latents,
signal_levels = self.max_steps - 1, signal_levels = self.max_steps - 1,
step_sizes = step_size, step_sizes = step_size,
@ -1958,7 +1958,7 @@ class DynamicsWorldModel(Module):
continuous_actions = continuous_actions, continuous_actions = continuous_actions,
latent_is_noised = True, latent_is_noised = True,
return_pred_only = True, return_pred_only = True,
return_agent_tokens = True return_intermediates = True
) )
agent_embed = agent_embed[..., agent_index, :] agent_embed = agent_embed[..., agent_index, :]

2
pyproject.toml
View File

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