Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-11-04 22:26:36 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity
9f90033ce4
Wan2.1/wan
History
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
..
configs Add CPU offload support for FSDP in Wan model 2025-07-22 21:03:47 +08:00
distributed Add CPU offload support for FSDP in Wan model 2025-07-22 21:03:47 +08:00
modules Format the code (#402) 2025-05-16 12:35:38 +08:00
utils Format the code (#402) 2025-05-16 12:35:38 +08:00
__init__.py Format the code (#402) 2025-05-16 12:35:38 +08:00
first_last_frame2video.py Format the code (#402) 2025-05-16 12:35:38 +08:00
image2video.py Add CPU offload support for FSDP in Wan model 2025-07-22 21:03:47 +08:00
text2video.py Format the code (#402) 2025-05-16 12:35:38 +08:00
vace.py Format the code (#402) 2025-05-16 12:35:38 +08:00