Update pipeline_stable_diffusion_xl_instantid_img2img.py

pipeline_stable_diffusion_xl_instantid_img2img.py

@@ -734,16 +734,16 @@ class StableDiffusionXLInstantIDImg2ImgPipeline(StableDiffusionXLControlNetImg2I
         controlnet = self.controlnet._orig_mod if is_compiled_module(self.controlnet) else self.controlnet
 
         # align format for control guidance
-        if isinstance(controlnet, MultiControlNetModel):
-            mult = len(controlnet.nets)
-        else:
-            mult = 1
-
-        if not isinstance(control_guidance_start, list):
-            control_guidance_start = mult * [control_guidance_start]
-
-        if not isinstance(control_guidance_end, list):
-            control_guidance_end = mult * [control_guidance_end]
+        if not isinstance(control_guidance_start, list) and isinstance(control_guidance_end, list):
+            control_guidance_start = len(control_guidance_end) * [control_guidance_start]
+        elif not isinstance(control_guidance_end, list) and isinstance(control_guidance_start, list):
+            control_guidance_end = len(control_guidance_start) * [control_guidance_end]
+        elif not isinstance(control_guidance_start, list) and not isinstance(control_guidance_end, list):
+            mult = len(controlnet.nets) if isinstance(controlnet, MultiControlNetModel) else 1
+            control_guidance_start, control_guidance_end = (
+                mult * [control_guidance_start],
+                mult * [control_guidance_end],
+            )
 
         # 1. Check inputs. Raise error if not correct
         self.check_inputs(