Add some annotations.

2026-02-05 18:53:22 -08:00
parent 9db48498de
commit d9fabcc5ef
4 changed files with 22 additions and 24 deletions
@@ -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
@@ -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"