From b1defa02d4d160e77c3d69d83212afd76de42e43 Mon Sep 17 00:00:00 2001
From: shiimizu <shiimizu@protonmail.com>
Date: Sat, 25 Nov 2023 18:26:51 -0800
Subject: [PATCH] Fix `nv` generator typo.

---
 smZNodes.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/smZNodes.py b/smZNodes.py
index 6a64ee6..e98e02a 100644
--- 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: