added proper noising of init_latent as reccommened by jackloomen (#24…

…1 (comment))
huggingface · Aug 24, 2022 · 721d55d · 721d55d
1 parent 78101f4
commit 721d55d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/inference/inpainting.py b/examples/inference/inpainting.py
@@ -162,7 +162,8 @@ def __call__(
             latents = self.scheduler.step(noise_pred, t, latents, **extra_step_kwargs)["prev_sample"]
 
             #masking
-            latents=init_latents_orig*mask+latents*(1-mask)
+            init_latents_proper = self.scheduler.add_noise(init_latents_orig, noise, t)
+            latents = ( init_latents_proper * mask ) + ( latents * (1-mask) )
 
         # scale and decode the image latents with vae
         latents = 1 / 0.18215 * latents