Unverified Commit 0e868811 authored by YiYi Xu's avatar YiYi Xu Committed by GitHub
Browse files

fix inpainting pipeline when providing initial latents (#3641)



* fix latents

* fix copies

---------
Co-authored-by: default avataryiyixuxu <yixu310@gmail,com>
parent f1d47433
...@@ -868,8 +868,8 @@ class StableDiffusionControlNetInpaintPipeline(DiffusionPipeline, TextualInversi ...@@ -868,8 +868,8 @@ class StableDiffusionControlNetInpaintPipeline(DiffusionPipeline, TextualInversi
# if pure noise then scale the initial latents by the Scheduler's init sigma # if pure noise then scale the initial latents by the Scheduler's init sigma
latents = latents * self.scheduler.init_noise_sigma if is_strength_max else latents latents = latents * self.scheduler.init_noise_sigma if is_strength_max else latents
else: else:
latents = latents.to(device) noise = latents.to(device)
latents = latents * self.scheduler.init_noise_sigma latents = noise * self.scheduler.init_noise_sigma
outputs = (latents,) outputs = (latents,)
......
...@@ -653,8 +653,8 @@ class StableDiffusionInpaintPipeline(DiffusionPipeline, TextualInversionLoaderMi ...@@ -653,8 +653,8 @@ class StableDiffusionInpaintPipeline(DiffusionPipeline, TextualInversionLoaderMi
# if pure noise then scale the initial latents by the Scheduler's init sigma # if pure noise then scale the initial latents by the Scheduler's init sigma
latents = latents * self.scheduler.init_noise_sigma if is_strength_max else latents latents = latents * self.scheduler.init_noise_sigma if is_strength_max else latents
else: else:
latents = latents.to(device) noise = latents.to(device)
latents = latents * self.scheduler.init_noise_sigma latents = noise * self.scheduler.init_noise_sigma
outputs = (latents,) outputs = (latents,)
......
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