Compare commits

...

2 Commits

Author SHA1 Message Date
Sayak Paul
26278dd000 Merge branch 'main' into fix-sdxl-controlnet-inpaint-compile 2024-03-27 07:26:28 +05:30
yiyixu
f4b25bbf07 fix 2024-03-26 09:22:14 +00:00

View File

@@ -1601,10 +1601,7 @@ class StableDiffusionXLControlNetInpaintPipeline(
1.0 - float(i / len(timesteps) < s or (i + 1) / len(timesteps) > e)
for s, e in zip(control_guidance_start, control_guidance_end)
]
if isinstance(self.controlnet, MultiControlNetModel):
controlnet_keep.append(keeps)
else:
controlnet_keep.append(keeps[0])
controlnet_keep.append(keeps if isinstance(controlnet, MultiControlNetModel) else keeps[0])
# 9. Prepare extra step kwargs. TODO: Logic should ideally just be moved out of the pipeline
height, width = latents.shape[-2:]