Merge c5a6d87db7 into c709fcf0e7

wan/first_last_frame2video.py

@@ -13,6 +13,7 @@ import numpy as np
 import torch
 import torch.cuda.amp as amp
 import torch.distributed as dist
+import torchvision
 import torchvision.transforms.functional as TF
 from tqdm import tqdm
 
@@ -206,7 +207,12 @@ class WanFLF2V:
                 round(last_frame_size[1] * last_frame_resize_ratio),
             ]
             # 2. center crop
-            last_frame = TF.center_crop(last_frame, last_frame_size)
+            transform = torchvision.transforms.Compose([
+                torchvision.transforms.Resize((last_frame_size[0], last_frame_size[1])),
+                torchvision.transforms.CenterCrop((first_frame_size[0], first_frame_size[1]))
+            ])
+
+            last_frame = transform(last_frame)
 
         max_seq_len = ((F - 1) // self.vae_stride[0] + 1) * lat_h * lat_w // (
             self.patch_size[1] * self.patch_size[2])