Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-11-04 14:16:57 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity
49 Commits 6 Branches 0 Tags 22 MiB
66e5815491
Commit Graph

3 Commits

Author SHA1 Message Date
ErrorDing
66e5815491 Change file permissions for start_master.sh and start_rank1.sh to make them executable. 2025-07-23 11:16:16 +08:00
ErrorDing
a51658f197 Update script usage examples in start_master.sh and start_rank1.sh to include --sample_steps parameter 2025-07-23 10:37:07 +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