AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

fix nois in ldm

Browse Source
This commit is contained in:
patil-suraj
2022-06-15 10:15:54 +02:00
parent ca94e36c97
commit d4c2bcf8a3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/diffusers/pipelines/pipeline_latent_diffusion.py
View File

@@ -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