quickjkee
/

swd_pipeline

Model card Files Files and versions Community

quickjkee commited on 14 days ago

Commit

179b120

·

verified ·

1 Parent(s): 8923a6a

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -246,7 +246,7 @@ class SwDPipeline(StableDiffusion3Pipeline):
                 x0_pred = (latents - sigma * noise_pred)
                 if scales and i + 1 < len(scales):
                     x0_pred = torch.nn.functional.interpolate(x0_pred, size=scales[i + 1], mode='bicubic')
-                noise = torch.randn(x0_pred.shape, generator=generator, device=device, dtype=x0_pred.dtype)
                 latents = (1 - sigma_next) * x0_pred + sigma_next * noise
                 if latents.dtype != latents_dtype:

                 x0_pred = (latents - sigma * noise_pred)
                 if scales and i + 1 < len(scales):
                     x0_pred = torch.nn.functional.interpolate(x0_pred, size=scales[i + 1], mode='bicubic')
+                noise = torch.randn(x0_pred.shape, generator=generator, dtype=x0_pred.dtype).to(x0_pred.device)
                 latents = (1 - sigma_next) * x0_pred + sigma_next * noise
                 if latents.dtype != latents_dtype: