Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-06-17 04:47:41 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge 57843a2111 into e5a741309d

Browse Source
This commit is contained in:
Feiteng 2025-05-17 12:53:21 +08:00 committed by GitHub
commit 0c3b8d2182
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
wan/image2video.py
View File

@ -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, frame_num, 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:]