Compare commits

..

2 Commits

Author SHA1 Message Date
Jeremy Jin-Young Kim
d3a0e7b077
Merge 76bceb2fe5 into c709fcf0e7 2025-05-15 13:01:27 +08:00
Zhen Han
c709fcf0e7
fix vace size (#397) 2025-05-14 22:01:45 +08:00
2 changed files with 2 additions and 1 deletions

View File

@ -210,7 +210,7 @@ class VACEInference:
[src_mask], [src_mask],
[src_ref_images], [src_ref_images],
num_frames=num_frames, 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) device=self.pipe.device)
video = self.pipe.generate( video = self.pipe.generate(
prompt, prompt,

View File

@ -200,6 +200,7 @@ class WanVace(WanT2V):
else: else:
raise NotImplementedError(f'image_size {image_size} is not supported') raise NotImplementedError(f'image_size {image_size} is not supported')
image_size = (image_size[1], image_size[0])
image_sizes = [] image_sizes = []
for i, (sub_src_video, sub_src_mask) in enumerate(zip(src_video, src_mask)): 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: if sub_src_mask is not None and sub_src_video is not None: