Add some annotations.

examples/inference/basic/basic_ltx2_base.py

@@ -38,15 +38,15 @@ def main() -> None:
     output_path = "outputs_video/ltx2_basic/output_ltx2_base_t2v_1088_1920_1.0.mp4"
     generator.generate_video(
         prompt=PROMPT,
-            output_path=output_path,
-            save_video=True,
-            num_frames=121,
-            height=1088,
-            width=1920,
-            fps=24,
-            negative_prompt=official_negative_prompt,
-            guidance_scale=3.0,
-        )
+        output_path=output_path,
+        save_video=True,
+        num_frames=121,
+        height=1088,
+        width=1920,
+        fps=24,
+        negative_prompt=official_negative_prompt,
+        guidance_scale=3.0,
+    )
     generator.shutdown()
 
 

run_ltx2_no_guidance_compare.py

@@ -138,9 +138,6 @@ PROMPT = (
     "The woman covers her face, on the verge of tears. The tone is deadpan, "
     "absurd, and quietly tragic.")
 
-PROMPT = (
-    "A curious raccoon peeks through a vibrant field of yellow sunflowers.")
-
 official_negative_prompt = (
     "blurry, out of focus, overexposed, underexposed, low contrast, washed out colors, excessive noise, "
     "grainy texture, poor lighting, flickering, motion blur, distorted proportions, unnatural skin tones, "
@@ -155,7 +152,6 @@ official_negative_prompt = (
     "inconsistent tone, cinematic oversaturation, stylized filters, or AI artifacts."
 )
 
-
 @dataclass
 class Setting:
     height: int

scripts/checkpoint_conversion/convert_ltx2_weights.py

@@ -1,6 +1,19 @@
 # SPDX-License-Identifier: Apache-2.0
 """
 Convert LTX-2 weights to FastVideo naming conventions and split by component.
+
+LTX 2 conversion requires two huggingface models:
+- LTX 2 model
+- Gemma model
+
+Example usage:
+    python scripts/checkpoint_conversion/convert_ltx2_weights.py \\
+        --source "<PATH_TO_LOCAL_REPO>/Lightricks/LTX-2/ltx-2-19b-dev.safetensors" \\
+        --output "converted_weights/ltx2-base" \\
+        --class-name "LTX2Transformer3DModel" \\
+        --pipeline-class-name "LTX2Pipeline" \\
+        --diffusers-version "0.33.0.dev0" \\
+        --gemma-path "<PATH_TO_LOCAL_REPO>/google/gemma-3-12b-it"
 """
 
 from __future__ import annotations

scripts/checkpoint_conversion/convert_ltx2_weights.sh

@@ -1,11 +0,0 @@
-#!/bin/bash
-# Simple wrapper for convert_ltx2_weights.py
-# Edit the values below directly
-
-python /home/d1su/codes/FastVideo/scripts/checkpoint_conversion/convert_ltx2_weights.py \
-    --source "<PATH_TO_LOCAL_REPO>/Lightricks/LTX-2/ltx-2-19b-dev.safetensors" \
-    --output "converted_weights/ltx2-base" \
-    --class-name "LTX2Transformer3DModel" \
-    --pipeline-class-name "LTX2Pipeline" \
-    --diffusers-version "0.33.0.dev0" \
-    --gemma-path "<PATH_TO_LOCAL_REPO>/google/gemma-3-12b-it"