This commit is contained in:
lucidrains 2025-10-29 13:04:02 -07:00
parent c4a3cb09d5
commit 35b87c4fa1
2 changed files with 2 additions and 2 deletions

View File

@ -2870,7 +2870,7 @@ class DynamicsWorldModel(Module):
# maybe store old actions
if store_old_action_unembeds:
acc_policy_embed = safe_cat((acc_policy_embed, policy_embed))
acc_policy_embed = safe_cat((acc_policy_embed, policy_embed), dim = 1)
# sample actions

View File

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