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-21 09:46:49 -07:00
parent 4a5465eeb6
commit e7d9766608
3 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
dreamer4/dreamer4.py
View File

@ -1251,6 +1251,7 @@ class AxialSpaceTimeTransformer(Module):
final_norm = True final_norm = True
): ):
super().__init__() super().__init__()
assert depth >= time_block_every, f'depth must be at least {time_block_every}'
# hyper connections # hyper connections

2
pyproject.toml
View File

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

2
tests/test_dreamer.py
View File

@ -483,7 +483,7 @@ def test_cache_generate():
max_steps = 64, max_steps = 64,
num_tasks = 4, num_tasks = 4,
num_latent_tokens = 4, num_latent_tokens = 4,
depth = 1, depth = 4,
num_spatial_tokens = 1, num_spatial_tokens = 1,
pred_orig_latent = True, pred_orig_latent = True,
num_discrete_actions = 4, num_discrete_actions = 4,