Update README.md
Browse files
README.md
CHANGED
|
@@ -12,4 +12,19 @@ image = image_processor(
|
|
| 12 |
)
|
| 13 |
```
|
| 14 |
|
| 15 |
-
for wan 2.2 5B, the default method is here https://huggingface.co/YiYiXu/WanImageProcessor
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
)
|
| 13 |
```
|
| 14 |
|
| 15 |
+
for wan 2.2 5B, the default method is here https://huggingface.co/YiYiXu/WanImageProcessor
|
| 16 |
+
|
| 17 |
+
this is the code to resize
|
| 18 |
+
|
| 19 |
+
```py
|
| 20 |
+
|
| 21 |
+
image = load_image(
|
| 22 |
+
"https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/astronaut.jpg"
|
| 23 |
+
)
|
| 24 |
+
max_area = 720 * 1280
|
| 25 |
+
aspect_ratio = image.height / image.width
|
| 26 |
+
mod_value = pipe.vae_scale_factor_spatial * pipe.transformer.config.patch_size[1]
|
| 27 |
+
height = round(np.sqrt(max_area * aspect_ratio)) // mod_value * mod_value
|
| 28 |
+
width = round(np.sqrt(max_area / aspect_ratio)) // mod_value * mod_value
|
| 29 |
+
image = image.resize((width, height))
|
| 30 |
+
```
|