Unverified Commit ead82fed authored by YiYi Xu's avatar YiYi Xu Committed by GitHub
Browse files

fix torch.compile for multi-controlnet of sdxl inpaint (#7476)



fix
Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
parent 45b42d12
...@@ -1601,10 +1601,7 @@ class StableDiffusionXLControlNetInpaintPipeline( ...@@ -1601,10 +1601,7 @@ class StableDiffusionXLControlNetInpaintPipeline(
1.0 - float(i / len(timesteps) < s or (i + 1) / len(timesteps) > e) 1.0 - float(i / len(timesteps) < s or (i + 1) / len(timesteps) > e)
for s, e in zip(control_guidance_start, control_guidance_end) for s, e in zip(control_guidance_start, control_guidance_end)
] ]
if isinstance(self.controlnet, MultiControlNetModel): controlnet_keep.append(keeps if isinstance(controlnet, MultiControlNetModel) else keeps[0])
controlnet_keep.append(keeps)
else:
controlnet_keep.append(keeps[0])
# 9. Prepare extra step kwargs. TODO: Logic should ideally just be moved out of the pipeline # 9. Prepare extra step kwargs. TODO: Logic should ideally just be moved out of the pipeline
height, width = latents.shape[-2:] height, width = latents.shape[-2:]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment