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

Compare commits

base: Zengtudor:4037cb07e1d6cc9d933be48c8870e16a9921b1a6
Branches Tags
Zengtudor:main
Zengtudor:dev/format
Zengtudor:dev/vace_seed
Zengtudor:dev/vace_dev
Zengtudor:dev/vace_size
Zengtudor:dev/vace
...
compare: Zengtudor:0b95ddb46de6dd21edd930d450e3d19be9b439b5
Branches Tags
Zengtudor:main
Zengtudor:dev/format
Zengtudor:dev/vace_seed
Zengtudor:dev/vace_dev
Zengtudor:dev/vace_size
Zengtudor:dev/vace

2 Commits
4037cb07e1 ... 0b95ddb46d

Author SHA1 Message Date
yupeng1111
0b95ddb46d
Merge c5a6d87db7 into ec902046f6 2025-05-28 11:39:09 +10: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])