Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-12-19 13:42:05 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Compare commits

base: Zengtudor:20808f5db9ca302f6585d92ff16ae741fe169a9a
Branches Tags
Zengtudor:main
Zengtudor:dev/format
...
compare: Zengtudor:2b68a0957f2c3bfb77bd0d7ca04acf0f6415b480
Branches Tags
Zengtudor:main
Zengtudor:dev/format

3 Commits
20808f5db9 ... 2b68a0957f

Author SHA1 Message Date
Jeremy Jin-Young Kim
2b68a0957f
Merge 76bceb2fe5 into 841fe5237b 2025-12-08 17:53:34 +11:00
Shiguang Ai
841fe5237b
When frame_num is changed from 81 the tensor does not match (#533) 2025-12-02 14:53:33 +08:00
root
76bceb2fe5 fix: enalbe prompt extend for flf2v 2025-05-02 05:10:43 +00:00
2 changed files with 2 additions and 2 deletions
Show Stats Expand all files Collapse all files

2
generate.py
View File

@ -312,7 +312,7 @@ def generate(args):
elif args.prompt_extend_method == "local_qwen": elif args.prompt_extend_method == "local_qwen":
prompt_expander = QwenPromptExpander( prompt_expander = QwenPromptExpander(
model_name=args.prompt_extend_model, model_name=args.prompt_extend_model,
is_vl="i2v" in args.task, is_vl="i2v" in args.task or "flf2v" in args.task,
device=rank) device=rank)
else: else:
raise NotImplementedError( raise NotImplementedError(

2
wan/image2video.py
View File

@ -207,7 +207,7 @@ class WanI2V:
generator=seed_g, generator=seed_g,
device=self.device) device=self.device)
msk = torch.ones(1, 81, lat_h, lat_w, device=self.device) msk = torch.ones(1, F, lat_h, lat_w, device=self.device)
msk[:, 1:] = 0 msk[:, 1:] = 0
msk = torch.concat([ msk = torch.concat([
torch.repeat_interleave(msk[:, 0:1], repeats=4, dim=1), msk[:, 1:] torch.repeat_interleave(msk[:, 0:1], repeats=4, dim=1), msk[:, 1:]