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

fix task embed in presence of multiple agent tokens

Browse Source
This commit is contained in:
lucidrains 2025-10-08 08:42:25 -07:00
parent 4c2ed100a3
commit b62c08be65
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
dreamer4/dreamer4.py
View File

@ -1519,7 +1519,7 @@ class DynamicsWorldModel(Module):
assert self.num_tasks > 0
task_embeds = self.task_embed(tasks)
agent_tokens = agent_tokens + task_embeds
agent_tokens = einx.add('b ... d, b d', agent_tokens, task_embeds)
# maybe evolution