oops

lucidrains · Oct 28, 2022 · 804c8ad · 804c8ad
1 parent c1b1236
commit 804c8ad
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/phenaki_pytorch/phenaki_pytorch.py b/phenaki_pytorch/phenaki_pytorch.py
@@ -1066,7 +1066,7 @@ def sample(
                     with torch.no_grad():
                         scores = self.critic(video_token_ids)
 
-                    noise = K * (uniform(scores.shape, device) - 0.5) * (steps_til_x0 / self.steps)
+                    noise = noise_K * (uniform(scores.shape, device) - 0.5) * (steps_til_x0 / self.steps)
                     scores = scores + noise
                 else:
                     scores = logits.gather(2, rearrange(pred_video_ids, '... -> ... 1'))

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'phenaki-pytorch',
   packages = find_packages(exclude=[]),
-  version = '0.0.8',
+  version = '0.0.9',
   license='MIT',
   description = 'Phenaki - Pytorch',
   author = 'Phil Wang',