mirror of
https://github.com/Wan-Video/Wan2.1.git
synced 2025-06-17 21:07:41 +00:00
Compare commits
2 Commits
ffb3130330
...
5abc467822
Author | SHA1 | Date | |
---|---|---|---|
|
5abc467822 | ||
|
57843a2111 |
@ -204,7 +204,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, frame_num, 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:]
|
||||||
|
Loading…
Reference in New Issue
Block a user