This commit is contained in:
Eugene Fairley 2025-07-11 17:27:43 -07:00
parent 37384d8d52
commit acd91d30a7

View File

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