This commit is contained in:
Eugene Fairley 2025-07-11 17:22:40 -07:00
parent 67c4e7d98d
commit 37384d8d52

View File

@ -544,7 +544,7 @@ class WanTrackToVideo:
lat_h = height // 8
lat_w = width // 8
msk = torch.ones(1, latent.shape[2], lat_h, lat_w, device=start_image.device)
msk = torch.ones(1, length - 1, lat_h, lat_w, device=start_image.device)
msk[:, 1:] = 0
# repeat first frame 4 times