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
76e9427657
Format the code ( #402 )
...
* isort the code
* format the code
* Add yapf config file
* Remove torch cuda memory profiler
2025-05-16 12:35:38 +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
Ang Wang
9fb8845009
Add more explanation about the context parallel ( #260 )
2025-03-20 17:04:49 +08:00
Alan May
82c18d91a1
fix: correct Chinese ISO language code typo ( #10 )
...
Fix incorrect Chinese language ISO code abbreviation
from CH to ZH
2025-03-03 17:04:13 +08:00
zyckk4
2ef4e2184e
Fix an error saving videos on Windows ( #128 )
2025-03-03 13:48:19 +08:00
WanX-Video-1
65386b2e03
init upload
2025-02-25 22:07:47 +08:00
