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 size

Browse Source
This commit is contained in:
hanzhn 2025-05-14 21:49:33 +08:00
parent c2c10fc8d0
commit 04a6cb7a3c
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: