Skip to content

Commit 0e86881

Browse files
yiyixuxuyiyixuxu
andauthored
fix inpainting pipeline when providing initial latents (huggingface#3641)
* fix latents * fix copies --------- Co-authored-by: yiyixuxu <yixu310@gmail,com>
1 parent f1d4743 commit 0e86881

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -868,8 +868,8 @@ def prepare_latents(
868868
# if pure noise then scale the initial latents by the Scheduler's init sigma
869869
latents = latents * self.scheduler.init_noise_sigma if is_strength_max else latents
870870
else:
871-
latents = latents.to(device)
872-
latents = latents * self.scheduler.init_noise_sigma
871+
noise = latents.to(device)
872+
latents = noise * self.scheduler.init_noise_sigma
873873

874874
outputs = (latents,)
875875

src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -653,8 +653,8 @@ def prepare_latents(
653653
# if pure noise then scale the initial latents by the Scheduler's init sigma
654654
latents = latents * self.scheduler.init_noise_sigma if is_strength_max else latents
655655
else:
656-
latents = latents.to(device)
657-
latents = latents * self.scheduler.init_noise_sigma
656+
noise = latents.to(device)
657+
latents = noise * self.scheduler.init_noise_sigma
658658

659659
outputs = (latents,)
660660

0 commit comments

Comments
 (0)