From 30a5acc39f77bd1aa5f6f2a34a67abd73071c412 Mon Sep 17 00:00:00 2001 From: YiYi Xu Date: Fri, 1 Sep 2023 06:51:59 -1000 Subject: [PATCH] fix a bug in sdxl-controlnet-img2img when using MultiControlNetModel (#4862) fix Co-authored-by: yiyixuxu --- .../pipelines/controlnet/pipeline_controlnet_sd_xl_img2img.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl_img2img.py b/src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl_img2img.py index 6e66bdcc44..43ee40857e 100644 --- a/src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl_img2img.py +++ b/src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl_img2img.py @@ -1177,6 +1177,7 @@ class StableDiffusionXLControlNetImg2ImgPipeline(DiffusionPipeline, TextualInver do_classifier_free_guidance=do_classifier_free_guidance, guess_mode=guess_mode, ) + height, width = control_image.shape[-2:] elif isinstance(controlnet, MultiControlNetModel): control_images = [] @@ -1196,9 +1197,9 @@ class StableDiffusionXLControlNetImg2ImgPipeline(DiffusionPipeline, TextualInver control_images.append(control_image_) control_image = control_images + height, width = control_image[0].shape[-2:] else: assert False - height, width = control_image.shape[-2:] # 5. Prepare timesteps self.scheduler.set_timesteps(num_inference_steps, device=device)