mirror of
https://github.com/Wan-Video/Wan2.1.git
synced 2025-12-19 05:32:07 +00:00
Compare commits
4 Commits
32a7533b85
...
668cc7360c
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
668cc7360c | ||
|
|
841fe5237b | ||
|
|
e3db544b89 | ||
|
|
4e7f021bd3 |
@ -14,3 +14,9 @@ imageio-ffmpeg
|
||||
flash_attn
|
||||
gradio>=5.0.0
|
||||
numpy>=1.23.5,<2
|
||||
diffusers
|
||||
ftfy
|
||||
transformers
|
||||
einops
|
||||
dashscope
|
||||
imageio-2.37.0
|
||||
|
||||
@ -207,7 +207,7 @@ class WanI2V:
|
||||
generator=seed_g,
|
||||
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 = torch.concat([
|
||||
torch.repeat_interleave(msk[:, 0:1], repeats=4, dim=1), msk[:, 1:]
|
||||
|
||||
Loading…
Reference in New Issue
Block a user