fix sample image generation without seed failed close #1726

kohya-ss · Oct 26, 2024 · 014064f · 014064f
1 parent 56bf761
commit 014064f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/library/sd3_train_utils.py b/library/sd3_train_utils.py
@@ -316,6 +316,8 @@ def do_sample(
     # noise = get_noise(seed, latent).to(device)
     if seed is not None:
         generator = torch.manual_seed(seed)
+    else:
+        generator = None
     noise = (
         torch.randn(latent.size(), dtype=torch.float32, layout=latent.layout, generator=generator, device="cpu")
         .to(latent.dtype)