Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-12-20 06:02:03 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Compare commits

base: Zengtudor:2ed18af30ac88458e68916b45b1c040290072c9e
Branches Tags
Zengtudor:main
Zengtudor:dev/format
...
compare: Zengtudor:b31cce7906a75ccf3fab717d8f39d127e6a17f74
Branches Tags
Zengtudor:main
Zengtudor:dev/format

2 Commits
2ed18af30a ... b31cce7906

Author SHA1 Message Date
yupeng1111
b31cce7906
Merge c5a6d87db7 into 8f7f6514f1 2025-06-19 12:26:41 -04:00
澎鹏
c5a6d87db7 fix frame size bug 2025-04-30 14:44:10 +08:00
1 changed files with 7 additions and 1 deletions
Show Stats Expand all files Collapse all files

8
wan/first_last_frame2video.py
View File

@ -13,6 +13,7 @@ import numpy as np
import torch import torch
import torch.cuda.amp as amp import torch.cuda.amp as amp
import torch.distributed as dist import torch.distributed as dist
import torchvision
import torchvision.transforms.functional as TF import torchvision.transforms.functional as TF
from tqdm import tqdm from tqdm import tqdm
@ -211,7 +212,12 @@ class WanFLF2V:
round(last_frame_size[1] * last_frame_resize_ratio), round(last_frame_size[1] * last_frame_resize_ratio),
] ]
# 2. center crop # 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 // ( max_seq_len = ((F - 1) // self.vae_stride[0] + 1) * lat_h * lat_w // (
self.patch_size[1] * self.patch_size[2]) self.patch_size[1] * self.patch_size[2])