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-18 08:07:06 -07:00
parent 5fc0022bbf
commit 911a1a8434
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
dreamer4/dreamer4.py
View File

@ -1859,7 +1859,7 @@ class DynamicsWorldModel(Module):
if isinstance(tasks, int): if isinstance(tasks, int):
tasks = torch.full((batch_size,), tasks, device = self.device) tasks = torch.full((batch_size,), tasks, device = self.device)
assert tasks.shape[0] == batch_size assert not exists(tasks) or tasks.shape[0] == batch_size
# get state latent shape # get state latent shape

2
pyproject.toml
View File

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