From cc364dbe47d51be244171872f1d7a5f8008365d0 Mon Sep 17 00:00:00 2001 From: yiyixuxu Date: Thu, 1 Jun 2023 19:30:20 +0000 Subject: [PATCH 1/2] fix latents --- .../stable_diffusion/pipeline_stable_diffusion_inpaint.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py index 534748c35363..9a6f5dbf9480 100644 --- a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py +++ b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py @@ -653,8 +653,8 @@ def prepare_latents( # 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 else: - latents = latents.to(device) - latents = latents * self.scheduler.init_noise_sigma + noise = latents.to(device) + latents = noise * self.scheduler.init_noise_sigma outputs = (latents,) From b89922880b67f45630ce861da00f7ee58fd671ad Mon Sep 17 00:00:00 2001 From: yiyixuxu Date: Thu, 1 Jun 2023 19:35:55 +0000 Subject: [PATCH 2/2] fix copies --- .../pipelines/controlnet/pipeline_controlnet_inpaint.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint.py b/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint.py index 821a93028c5d..8ebcac2589a3 100644 --- a/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint.py +++ b/src/diffusers/pipelines/controlnet/pipeline_controlnet_inpaint.py @@ -868,8 +868,8 @@ def prepare_latents( # 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 else: - latents = latents.to(device) - latents = latents * self.scheduler.init_noise_sigma + noise = latents.to(device) + latents = noise * self.scheduler.init_noise_sigma outputs = (latents,)