Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-06-03 22:04:53 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix vace size

Browse Source
This commit is contained in:
wangang.wa 2025-05-14 21:57:36 +08:00
parent 18d53feb7a
commit ce3e15c1e5
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
gradio/vace.py
View File

@ -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,

1
wan/vace.py
View File

@ -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: