From d4c2bcf8a385b54809b3d14026e1fc26622b9132 Mon Sep 17 00:00:00 2001 From: patil-suraj Date: Wed, 15 Jun 2022 10:15:54 +0200 Subject: [PATCH] fix nois in ldm --- src/diffusers/pipelines/pipeline_latent_diffusion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/pipeline_latent_diffusion.py b/src/diffusers/pipelines/pipeline_latent_diffusion.py index fe1ed4ca30..88751312cf 100644 --- a/src/diffusers/pipelines/pipeline_latent_diffusion.py +++ b/src/diffusers/pipelines/pipeline_latent_diffusion.py @@ -944,7 +944,7 @@ class LatentDiffusion(DiffusionPipeline): # 3. optionally sample variance variance = 0 if eta > 0: - noise = self.noise_scheduler.sample_noise(image.shape, device=image.device, generator=generator) + noise = torch.randn(image.shape, generator=generator, device=image.device) variance = self.noise_scheduler.get_variance(t, num_inference_steps).sqrt() * eta * noise # 4. set current image to prev_image: x_t -> x_t-1