Update pipeline_controlnet_sd_xl_img2img.py

CyberVy · web-flow · commit 9322c1f1aac7 · 2025-02-24T00:02:05.000+08:00
diff --git a/src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl_img2img.py b/src/diffusers/pipelines/controlnet/pipeline_controlnet_sd_xl_img2img.py
@@ -1615,6 +1615,7 @@ def __call__(
                     )
                     add_time_ids = callback_outputs.pop("add_time_ids", add_time_ids)
                     add_neg_time_ids = callback_outputs.pop("add_neg_time_ids", add_neg_time_ids)
+                    control_image = callback_outputs.pop("control_image", control_image)
 
                 # call the callback, if provided
                 if i == len(timesteps) - 1 or ((i + 1) > num_warmup_steps and (i + 1) % self.scheduler.order == 0):

Original file line number	Diff line number	Diff line change
`@@ -1615,6 +1615,7 @@ def __call__(`
`1615`	`1615`	`)`
`1616`	`1616`	`add_time_ids = callback_outputs.pop("add_time_ids", add_time_ids)`
`1617`	`1617`	`add_neg_time_ids = callback_outputs.pop("add_neg_time_ids", add_neg_time_ids)`
	`1618`	`+ control_image = callback_outputs.pop("control_image", control_image)`
`1618`	`1619`
`1619`	`1620`	`# call the callback, if provided`
`1620`	`1621`	`if i == len(timesteps) - 1 or ((i + 1) > num_warmup_steps and (i + 1) % self.scheduler.order == 0):`