Update pipeline_stable_diffusion_controlnet.py (huggingface#2917)

Beyondo · Mar 31, 2023 · 7447f75 · 7447f75
1 parent a5bdb67
commit 7447f75
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_controlnet.py b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_controlnet.py
@@ -698,15 +698,15 @@ def _default_height_width(self, height, width, image):
             if isinstance(image, PIL.Image.Image):
                 height = image.height
             elif isinstance(image, torch.Tensor):
-                height = image.shape[3]
+                height = image.shape[2]
 
             height = (height // 8) * 8  # round down to nearest multiple of 8
 
         if width is None:
             if isinstance(image, PIL.Image.Image):
                 width = image.width
             elif isinstance(image, torch.Tensor):
-                width = image.shape[2]
+                width = image.shape[3]
 
             width = (width // 8) * 8  # round down to nearest multiple of 8