fix size

README.md

@@ -1,5 +1,5 @@
 # Wan2.1
-
+ 
 <p align="center">
     <img src="assets/logo.png" width="400"/>
 <p>

gradio/vace.py

@@ -210,7 +210,7 @@ class VACEInference:
                                                                          [src_mask],
                                                                          [src_ref_images],
                                                                          num_frames=num_frames,
-                                                                         image_size=SIZE_CONFIGS[f"{output_height}*{output_width}"],
+                                                                         image_size=SIZE_CONFIGS[f"{output_width}*{output_height}"],
                                                                          device=self.pipe.device)
         video = self.pipe.generate(
             prompt,

wan/vace.py

@@ -200,6 +200,7 @@ class WanVace(WanT2V):
         else:
             raise NotImplementedError(f'image_size {image_size} is not supported')
 
+        image_size = (image_size[1], image_size[0])
         image_sizes = []
         for i, (sub_src_video, sub_src_mask) in enumerate(zip(src_video, src_mask)):
             if sub_src_mask is not None and sub_src_video is not None: