fix: update _get_bucket_size to vae_image_processor.get_default_height_width
Browse files- inference.py +1 -1
inference.py
CHANGED
|
@@ -57,7 +57,7 @@ def main():
|
|
| 57 |
|
| 58 |
images = [Image.open(p).convert("RGB") for p in args.images]
|
| 59 |
|
| 60 |
-
target_h, target_w = pipe.
|
| 61 |
|
| 62 |
result = pipe(
|
| 63 |
images=images,
|
|
|
|
| 57 |
|
| 58 |
images = [Image.open(p).convert("RGB") for p in args.images]
|
| 59 |
|
| 60 |
+
target_h, target_w = pipe.vae_image_processor.get_default_height_width(images[-1])
|
| 61 |
|
| 62 |
result = pipe(
|
| 63 |
images=images,
|