Update pipeline_stable_diffusion_controlnet.py (#2917)

7447f75b · Patrick von Platen · GitHub · a5bdb678 · 7447f75b
Unverified Commit 7447f75b authored Mar 31, 2023 by Patrick von Platen Committed by GitHub Mar 31, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_controlnet.py .../stable_diffusion/pipeline_stable_diffusion_controlnet.py +2 -2

No files found.
--- a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_controlnet.py
+++ b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_controlnet.py
@@ -698,7 +698,7 @@ class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInversionLoade
            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

@@ -706,7 +706,7 @@ class StableDiffusionControlNetPipeline(DiffusionPipeline, TextualInversionLoade
            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