[Feature] Offload all text encoders by default (#594)

This commit is contained in:
Wenxuan Tan
2025-07-03 19:30:14 -05:00
committed by GitHub
parent b9423ca3f8
commit ed1e8d6bad
43 changed files with 357 additions and 256 deletions
+1 -1
View File
@@ -10,7 +10,7 @@ def main():
# attempt to identify the optimal arguments.
generator = VideoGenerator.from_pretrained(
"Wan-AI/Wan2.1-T2V-1.3B-Diffusers",
# if num_gpus > 1, FastVideo will automatically handle distributed setup
# FastVideo will automatically handle distributed setup
num_gpus=2,
use_fsdp_inference=True,
use_cpu_offload=False