Skip to content

GitLab

Unverified Commit 0edb4cac authored by Ella Charlaix's avatar Ella Charlaix Committed by GitHub
Browse files

Fix image processor inputs width (#4853) 

fix width for np array inputs
parent 85b3f08c
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/diffusers/image_processor.py
View file @ 0edb4cac
...@@ -192,7 +192,7 @@ class VaeImageProcessor(ConfigMixin): ...@@ -192,7 +192,7 @@ class VaeImageProcessor(ConfigMixin):
elif isinstance(image, torch.Tensor): elif isinstance(image, torch.Tensor):
width = image.shape[3] width = image.shape[3]
else: else:
height = image.shape[2] width = image.shape[2]
width, height = ( width, height = (
x - x % self.config.vae_scale_factor for x in (width, height) x - x % self.config.vae_scale_factor for x in (width, height)
......
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