Fix nv generator typo.

shiimizu · Nov 26, 2023 · b1defa0 · b1defa0
1 parent b567095
commit b1defa0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/smZNodes.py b/smZNodes.py
@@ -610,7 +610,7 @@ def get_generator(seed):
     for i in range(unique_inds[-1]+1):
         shape = [1] + list(latent_image.size())[1:]
         if opts.randn_source == 'nv':
-            noise = torch.asarray(rng.randn(shape), device=devices.cpu)
+            noise = torch.asarray(generator.randn(shape), device=devices.cpu)
         else:
             noise = torch.randn(shape, dtype=latent_image.dtype, layout=latent_image.layout, device=device, generator=generator)
         if i in unique_inds: