make sure readme runs

lucidrains · Dec 10, 2022 · 8b7be7f · 8b7be7f
1 parent 8de344e
commit 8b7be7f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -109,7 +109,8 @@ unet = SpaceTimeUnet(
     channels = 3,
     dim_mult = (1, 2, 4, 8),
     temporal_compression = (False, False, False, True),
-    self_attns = (False, False, False, True)
+    self_attns = (False, False, False, True),
+    condition_on_timestep = False
 ).cuda()
 
 # train on images