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-12-03 08:11:47 -08:00
parent fb8c3793b4
commit 9efe269688
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

@ -2762,7 +2762,7 @@ class DynamicsWorldModel(Module):
): ):
with world_model_forward_context(): with world_model_forward_context():
_, (agent_embeds, _) = self.forward( _, (embeds, _) = 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,
@ -2774,7 +2774,7 @@ class DynamicsWorldModel(Module):
return_intermediates = True return_intermediates = True
) )
agent_embeds = agent_embeds[..., agent_index, :] agent_embeds = embeds.agent[..., agent_index, :]
# maybe detach agent embed # maybe detach agent embed

2
pyproject.toml
View File

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