image

hlky · hlky · commit cbf74b0478a2 · 2024-12-05T13:38:35.000Z
diff --git a/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_sd_xl.py b/src/diffusers/pipelines/controlnet/pipeline_controlnet_union_sd_xl.py
@@ -1272,7 +1272,7 @@ def __call__(
                     do_classifier_free_guidance=self.do_classifier_free_guidance,
                     guess_mode=guess_mode,
                 )
-                height, width = image.shape[-2:]
+                height, width = image[image_type].shape[-2:]
 
         # 5. Prepare timesteps
         timesteps, num_inference_steps = retrieve_timesteps(

Original file line number	Diff line number	Diff line change
`@@ -1272,7 +1272,7 @@ def __call__(`
`1272`	`1272`	`do_classifier_free_guidance=self.do_classifier_free_guidance,`
`1273`	`1273`	`guess_mode=guess_mode,`
`1274`	`1274`	`)`
`1275`		`- height, width = image.shape[-2:]`
	`1275`	`+ height, width = image[image_type].shape[-2:]`
`1276`	`1276`
`1277`	`1277`	`# 5. Prepare timesteps`
`1278`	`1278`	`timesteps, num_inference_steps = retrieve_timesteps(`