Compare commits

...

2 Commits

Author SHA1 Message Date
Feiteng
5abc467822
Merge 57843a2111 into c709fcf0e7 2025-05-14 22:05:47 +08:00
Feiteng
57843a2111 Fix msk frame_num 2025-05-09 07:41:18 +00:00

View File

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