Compare commits

..

1 Commits

Author SHA1 Message Date
Vibhanshu Kumar Shubham
32a7533b85
Merge e3db544b89 into 7c81b2f27d 2025-07-18 09:12:38 -07:00

View File

@ -207,7 +207,7 @@ class WanI2V:
generator=seed_g,
device=self.device)
msk = torch.ones(1, F, lat_h, lat_w, device=self.device)
msk = torch.ones(1, 81, 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:]