[feat!] Disable FSDP inference by default (#1001)

This commit is contained in:
XOR-op
2026-01-13 14:20:05 -08:00
committed by GitHub
parent 23a4531491
commit b7f69c2c1d
18 changed files with 22 additions and 17 deletions
+1 -1
View File
@@ -14,7 +14,7 @@ def main():
model_name,
# FastVideo will automatically handle distributed setup
num_gpus=1,
use_fsdp_inference=True,
use_fsdp_inference=False, # set to True if GPU is out of memory
# Adjust these offload parameters if you have < 32GB of VRAM
text_encoder_cpu_offload=True,
pin_cpu_memory=True, # set to false if low CPU RAM or hit obscure "CUDA error: Invalid argument"