Compare commits

...

2 Commits

Author SHA1 Message Date
hanzhn
04a6cb7a3c fix size 2025-05-14 21:49:33 +08:00
hanzhen.hz
c2c10fc8d0 update 2025-05-14 15:50:17 +08:00
3 changed files with 3 additions and 2 deletions

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,

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: