Commit Graph

5 Commits

Author SHA1 Message Date
ErrorDing
dc96cbb091 Expand supported resolutions in SIZE_CONFIGS and MAX_AREA_CONFIGS; update SUPPORTED_SIZES for i2v-14B to include new dimensions. 2025-07-23 11:14:55 +08:00
ErrorDing
9f90033ce4 Add CPU offload support for FSDP in Wan model
- Introduced `--cpu_offload` argument in `generate.py` for enabling CPU offload.
- Updated `WanI2V` class in `image2video.py` to handle CPU offload during model initialization and sharding.
- Added new functions in `fsdp.py` for CPU initialization and sharding with CPU offload.
- Expanded supported sizes in `configs/__init__.py` to include additional resolutions.
2025-07-22 21:03:47 +08:00
Ang Wang
18d53feb7a
[feature] Add VACE (#389)
* Add VACE

* Support training with multiple gpus

* Update default args for vace task

* vace block update

* Add vace exmaple jpg

* Fix dist vace fwd hook error

* Update vace exmample

* Update vace args

* Update pipeline name for vace

* vace gradio and Readme

* Update vace snake png

---------

Co-authored-by: hanzhn <han.feng.jason@gmail.com>
2025-05-14 20:44:25 +08:00
yupeng1111
df44622e72
[feature] Wan2.1-FLF2V-14B (#338)
Co-authored-by: 澎鹏 <shiyupeng.syp@taobao.com>
2025-04-17 21:56:46 +08:00
WanX-Video-1
65386b2e03 init upload 2025-02-25 22:07:47 +08:00