scale image

This commit is contained in:
Eugene Fairley 2025-06-27 13:34:22 -07:00
parent 7899fb868c
commit 78c52aac5f

View File

@ -656,7 +656,7 @@ class WanTrackToVideo:
zero_frames = torch.zeros(3, 81 - 1, height, width)
start_image = start_image.permute(3,0,1,2) # C, T, H, W
start_image = start_image.permute(3,0,1,2) * 2 - 1 # C, T, H, W
res = torch.concat([
start_image.to(start_image.device),
zero_frames