fix device for ldm

2026-01-29 07:22:12 +03:00 · 2022-06-15 10:17:36 +02:00
parent a3784522a8
commit cdb3c4931b
1 changed files with 1 additions and 2 deletions
--- a/src/diffusers/pipelines/pipeline_latent_diffusion.py
+++ b/src/diffusers/pipelines/pipeline_latent_diffusion.py
@@ -902,9 +902,8 @@ class LatentDiffusion(DiffusionPipeline):

        image = torch.randn(
            (batch_size, self.unet.in_channels, self.unet.image_size, self.unet.image_size),
-            device=torch_device,
            generator=generator,
-        )
+        ).to(torch_device)

        # See formulas (12) and (16) of DDIM paper https://arxiv.org/pdf/2010.02502.pdf
        # Ideally, read DDIM paper in-detail understanding