Compare commits

...

4 Commits

Author SHA1 Message Date
Vibhanshu Kumar Shubham
668cc7360c
Merge e3db544b89 into 841fe5237b 2025-12-09 01:31:05 +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
Vibhanshu Kumar Shubham
e3db544b89
Merge branch 'main' into patch-2 2025-07-09 17:18:57 +05:30
Vibhanshu Kumar Shubham
4e7f021bd3
Update requirements.txt
here iam adding some requirements not mentioned in requireme
2025-07-01 14:43:07 +05:30
2 changed files with 7 additions and 1 deletions

View File

@ -14,3 +14,9 @@ imageio-ffmpeg
flash_attn flash_attn
gradio>=5.0.0 gradio>=5.0.0
numpy>=1.23.5,<2 numpy>=1.23.5,<2
diffusers
ftfy
transformers
einops
dashscope
imageio-2.37.0

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:]