mirror of
https://github.com/Wan-Video/Wan2.1.git
synced 2025-12-19 13:42:05 +00:00
Compare commits
3 Commits
4037cb07e1
...
b31cce7906
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
b31cce7906 | ||
|
|
8f7f6514f1 | ||
|
|
c5a6d87db7 |
@ -36,6 +36,8 @@ In this repository, we present **Wan2.1**, a comprehensive and open suite of vid
|
|||||||
|
|
||||||
## Community Works
|
## Community Works
|
||||||
If your work has improved **Wan2.1** and you would like more people to see it, please inform us.
|
If your work has improved **Wan2.1** and you would like more people to see it, please inform us.
|
||||||
|
- [HyperMotion](https://vivocameraresearch.github.io/hypermotion/), a human image animation framework based on **Wan2.1**, addresses the challenge of generating complex human body motions in pose-guided animation. Refer to [their website](https://vivocameraresearch.github.io/magictryon/) for more examples.
|
||||||
|
- [MagicTryOn](https://vivocameraresearch.github.io/magictryon/), a video virtual try-on framework built upon **Wan2.1-14B-I2V**, addresses the limitations of existing models in expressing garment details and maintaining dynamic stability during human motion. Refer to [their website](https://vivocameraresearch.github.io/magictryon/) for more examples.
|
||||||
- [ATI](https://github.com/bytedance/ATI), built on **Wan2.1-I2V-14B**, is a trajectory-based motion-control framework that unifies object, local, and camera movements in video generation. Refer to [their website](https://anytraj.github.io/) for more examples.
|
- [ATI](https://github.com/bytedance/ATI), built on **Wan2.1-I2V-14B**, is a trajectory-based motion-control framework that unifies object, local, and camera movements in video generation. Refer to [their website](https://anytraj.github.io/) for more examples.
|
||||||
- [Phantom](https://github.com/Phantom-video/Phantom) has developed a unified video generation framework for single and multi-subject references based on both **Wan2.1-T2V-1.3B** and **Wan2.1-T2V-14B**. Please refer to [their examples](https://github.com/Phantom-video/Phantom).
|
- [Phantom](https://github.com/Phantom-video/Phantom) has developed a unified video generation framework for single and multi-subject references based on both **Wan2.1-T2V-1.3B** and **Wan2.1-T2V-14B**. Please refer to [their examples](https://github.com/Phantom-video/Phantom).
|
||||||
- [UniAnimate-DiT](https://github.com/ali-vilab/UniAnimate-DiT), based on **Wan2.1-14B-I2V**, has trained a Human image animation model and has open-sourced the inference and training code. Feel free to enjoy it!
|
- [UniAnimate-DiT](https://github.com/ali-vilab/UniAnimate-DiT), based on **Wan2.1-14B-I2V**, has trained a Human image animation model and has open-sourced the inference and training code. Feel free to enjoy it!
|
||||||
|
|||||||
@ -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])
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user