Skip to content

GitLab

Commit cdb3c493 authored by patil-suraj's avatar patil-suraj
Browse files

fix device for ldm

parent a3784522
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
src/diffusers/pipelines/pipeline_latent_diffusion.py
View file @ cdb3c493
...@@ -902,9 +902,8 @@ class LatentDiffusion(DiffusionPipeline): ...@@ -902,9 +902,8 @@ class LatentDiffusion(DiffusionPipeline):
image = torch.randn( image = torch.randn(
(batch_size, self.unet.in_channels, self.unet.image_size, self.unet.image_size), (batch_size, self.unet.in_channels, self.unet.image_size, self.unet.image_size),
device=torch_device,
generator=generator, generator=generator,
) ).to(torch_device)
# See formulas (12) and (16) of DDIM paper https://arxiv.org/pdf/2010.02502.pdf # See formulas (12) and (16) of DDIM paper https://arxiv.org/pdf/2010.02502.pdf
# Ideally, read DDIM paper in-detail understanding # Ideally, read DDIM paper in-detail understanding
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment