debugging i2v

2025-08-01 21:53:26 -07:00
parent 7b3c4fbdd1
commit 42da276006
3 changed files with 24 additions and 3 deletions
@@ -14,9 +14,9 @@ def main():
        # FastVideo will automatically handle distributed setup
        num_gpus=1,
        use_fsdp_inference=True,
-        dit_cpu_offload=False,
+        dit_cpu_offload=True,
        vae_cpu_offload=False,
-        text_encoder_cpu_offload=False,
+        text_encoder_cpu_offload=True,
        # Set pin_cpu_memory to false if CPU RAM is limited and there're no frequent CPU-GPU transfer
        pin_cpu_memory=True,
        # image_encoder_cpu_offload=False,