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

fix vace size (#397)

Browse Source
This commit is contained in:
Zhen Han 2025-05-14 22:01:45 +08:00 committed by GitHub
parent 18d53feb7a
commit c709fcf0e7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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: