fix sample image generation without seed failed close #1726

2026-04-08 22:35:09 +00:00 · 2024-10-26 18:59:45 +09:00
parent 56bf761164
commit 014064fd81
1 changed files with 2 additions and 0 deletions
--- 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)