52 lines
1.7 KiB
Python
52 lines
1.7 KiB
Python
# 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."
|
|
)
|
|
|
|
# HF model ID (downloaded automatically). Distilled repos include upsamplers + refine LoRA reference.
|
|
MODEL_ID = "FastVideo/LTX2-Distilled-Diffusers"
|
|
|
|
OUTPUT_PATH = "outputs_video/ltx2_upscale/ltx2_two_stage.mp4"
|
|
|
|
|
|
def main() -> None:
|
|
generator = VideoGenerator.from_pretrained(
|
|
MODEL_ID,
|
|
num_gpus=8,
|
|
ltx2_refine_enabled=True,
|
|
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,
|
|
# DistilledPipeline uses the 8-step distilled schedule without CFG.
|
|
num_inference_steps=8,
|
|
guidance_scale=1.0,
|
|
save_video=True,
|
|
)
|
|
generator.shutdown()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|