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
......@@ -192,7 +192,7 @@ class VaeImageProcessor(ConfigMixin):
elif isinstance(image, torch.Tensor):
width = image.shape[3]
else:
height = image.shape[2]
width = image.shape[2]
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