Merge pull request #32 from JCBrouwer/patch-1

Fix offset noise
ExponentialML · Apr 3, 2023 · 25697f9 · 25697f9
2 parents 9cfcae5 + 9c47acb
commit 25697f9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -203,7 +203,7 @@ def sample_noise(latents, noise_strength, use_offset_noise):
 
     if use_offset_noise:
         offset_noise = torch.randn(b, c, f, 1, 1, device=latents.device)
-        noise = noise_latents + noise_strength * offset_noise
+        noise_latents = noise_latents + noise_strength * offset_noise
 
     return noise_latents