[Refactor] SP Mask --> original seq len; HunyuanVideo 1.5 does not need mask (#1142)

2026-03-04 11:44:47 +08:00
parent e63b7d8ac4
commit cd1b7cf139
17 changed files with 154 additions and 360 deletions
@@ -11,7 +11,7 @@ def main():
    generator = VideoGenerator.from_pretrained(
        "Wan-AI/Wan2.2-T2V-A14B-Diffusers",
        # FastVideo will automatically handle distributed setup
-        num_gpus=1,
+        num_gpus=2,
        use_fsdp_inference=False, # set to True if GPU is out of memory 
        dit_cpu_offload=True, # DiT need to be offloaded for MoE
        vae_cpu_offload=False,