4 Commits

Author	SHA1	Message	Date
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
Wenting Shen	bc3249d61c	add free stroage api for FSDP (#277) ... Signed-off-by: wenting.swt <wenting.swt@alibaba-inc.com>	2025-03-26 15:49:37 +08:00
WanX-Video-1	65386b2e03	init upload	2025-02-25 22:07:47 +08:00