test

examples/inference/basic/basic_ltx2_upscale.py

@@ -0,0 +1,53 @@
+# SPDX-License-Identifier: Apache-2.0
+from fastvideo import VideoGenerator
+
+PROMPT = (
+    "A warm sunny backyard. The camera starts in a tight cinematic close-up "
+    "of a woman and a man in their 30s, facing each other with serious "
+    "expressions. The woman, emotional and dramatic, says softly, \"That's "
+    "it... Dad's lost it. And we've lost Dad.\" The man exhales, slightly "
+    "annoyed: \"Stop being so dramatic, Jess.\" A beat. He glances aside, "
+    "then mutters defensively, \"He's just having fun.\" The camera slowly "
+    "pans right, revealing the grandfather in the garden wearing enormous "
+    "butterfly wings, waving his arms in the air like he's trying to take "
+    "off. He shouts, \"Wheeeew!\" as he flaps his wings with full commitment. "
+    "The woman covers her face, on the verge of tears. The tone is deadpan, "
+    "absurd, and quietly tragic."
+)
+
+# FastVideo-converted weights (align settings to the official distilled pipeline).
+MODEL_PATH = "converted/ltx2_diffusers"
+UPSAMPLER_PATH = "converted/ltx2_spatial_upscaler"
+
+OUTPUT_PATH = "outputs_video/ltx2_upscale/ltx2_official_two_stage2222.mp4"
+
+
+def main() -> None:
+    generator = VideoGenerator.from_pretrained(
+        MODEL_PATH,
+        num_gpus=1,
+        ltx2_refine_enabled=True,
+        ltx2_refine_upsampler_path=UPSAMPLER_PATH,
+        ltx2_refine_num_inference_steps=3,
+        ltx2_refine_guidance_scale=1.0,
+        ltx2_refine_add_noise=True,
+    )
+
+    generator.generate_video(
+        prompt=PROMPT,
+        output_path=OUTPUT_PATH,
+        height=960,
+        width=1664,
+        num_frames=81,
+        fps=24,
+        seed=10,
+        # Distilled schedule in LTX-2 uses 8 steps; setting this keeps parity.
+        num_inference_steps=8,
+        guidance_scale=1.0,
+        save_video=True,
+    )
+    generator.shutdown()
+
+
+if __name__ == "__main__":
+    main()