From ce3e15c1e5168cd6a072c632e449e41ab809b314 Mon Sep 17 00:00:00 2001
From: "wangang.wa" <wangang.wa@alibaba-inc.com>
Date: Wed, 14 May 2025 21:57:36 +0800
Subject: [PATCH] Fix vace size

---
 gradio/vace.py | 2 +-
 wan/vace.py    | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/gradio/vace.py b/gradio/vace.py
index 06c4b0d..75f780a 100644
--- a/gradio/vace.py
+++ b/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,
diff --git a/wan/vace.py b/wan/vace.py
index 1b94e88..d792e9b 100644
--- a/wan/vace.py
+++ b/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: