Update app.py

app.py

@@ -12,12 +12,15 @@ from typing import Optional
 from huggingface_hub import hf_hub_download
 from ltx_pipelines.distilled import DistilledPipeline
 from ltx_core.tiling import TilingConfig
+from ltx_core.loader.primitives import LoraPathStrengthAndSDOps
+from ltx_core.loader.sd_ops import LTXV_LORA_COMFY_RENAMING_MAP
 from ltx_pipelines.constants import (
     DEFAULT_SEED,
     DEFAULT_HEIGHT,
     DEFAULT_WIDTH,
     DEFAULT_NUM_FRAMES,
     DEFAULT_FRAME_RATE,
+    DEFAULT_LORA_STRENGTH,
 )
 
 # Default prompt from docstring example
@@ -26,7 +29,8 @@ DEFAULT_PROMPT = "An astronaut hatches from a fragile egg on the surface of the
 # HuggingFace Hub defaults
 DEFAULT_REPO_ID = "LTX-Colab/LTX-Video-Preview"
 DEFAULT_GEMMA_REPO_ID = "google/gemma-3-12b-it-qat-q4_0-unquantized"
-DEFAULT_CHECKPOINT_FILENAME = "ltx-2-19b-distilled-rc1.safetensors"
+DEFAULT_CHECKPOINT_FILENAME = "ltx-2-19b-dev-rc1.safetensors"
+DEFAULT_DISTILLED_LORA_FILENAME = "ltx-2-19b-distilled-lora-384-rc1.safetensors"
 DEFAULT_SPATIAL_UPSAMPLER_FILENAME = "ltx-2-spatial-upscaler-x2-1.0-rc1.safetensors"
 
 def get_hub_or_local_checkpoint(repo_id: Optional[str] = None, filename: Optional[str] = None):
@@ -52,18 +56,29 @@ print("Loading LTX-2 Distilled pipeline...")
 print("=" * 80)
 
 checkpoint_path = get_hub_or_local_checkpoint(DEFAULT_REPO_ID, DEFAULT_CHECKPOINT_FILENAME)
+distilled_lora_path = get_hub_or_local_checkpoint(DEFAULT_REPO_ID, DEFAULT_DISTILLED_LORA_FILENAME)
 spatial_upsampler_path = get_hub_or_local_checkpoint(DEFAULT_REPO_ID, DEFAULT_SPATIAL_UPSAMPLER_FILENAME)
 
 print(f"Initializing pipeline with:")
 print(f"  checkpoint_path={checkpoint_path}")
+print(f"  distilled_lora_path={distilled_lora_path}")
 print(f"  spatial_upsampler_path={spatial_upsampler_path}")
 print(f"  gemma_root={DEFAULT_GEMMA_REPO_ID}")
 
+# Load distilled LoRA as a regular LoRA
+loras = [
+    LoraPathStrengthAndSDOps(
+        path=distilled_lora_path,
+        strength=DEFAULT_LORA_STRENGTH,
+        sd_ops=LTXV_LORA_COMFY_RENAMING_MAP,
+    )
+]
+
 pipeline = DistilledPipeline(
     checkpoint_path=checkpoint_path,
     spatial_upsampler_path=spatial_upsampler_path,
     gemma_root=DEFAULT_GEMMA_REPO_ID,
-    loras=[],
+    loras=loras,
     fp8transformer=False,
 )
 
@@ -224,4 +239,4 @@ with gr.Blocks(title="LTX-2 Distilled Image-to-Video") as demo:
 
 
 if __name__ == "__main__":
-    demo.launch(theme=gr.themes.Citrus(), share=True)
+    demo.launch(theme=gr.themes.Citrus())