Update app.py

app.py

@@ -260,6 +260,7 @@ checkpoint_path = hf_hub_download(
     local_dir=str(weights_dir),
     local_dir_use_symlinks=False,
 )
+spatial_upsampler_path = hf_hub_download(repo_id=LTX_MODEL_REPO, filename="ltx-2.3-spatial-upscaler-x2-1.0.safetensors")
 
 print("[Gemma] Setting up abliterated Gemma text encoder...")
 MERGED_WEIGHTS = "/tmp/abliterated_gemma_merged.safetensors"
@@ -361,12 +362,14 @@ print(f"Realism LoRA: {realism_lora_path}")
 print(f"Transition LoRA: {transition_lora_path}")
 # ----------------------------------------------------------------
 
+print(f"Spatial upsampler: {spatial_upsampler_path}")
 print(f"Checkpoint: {checkpoint_path}")
 
 # Initialize pipeline WITH text encoder and optional audio support
 # ---- Replace block (pipeline init) lines 275-281 ----
 pipeline = LTX23DistilledA2VPipeline(
     distilled_checkpoint_path=checkpoint_path,
+    spatial_upsampler_path=spatial_upsampler_path,
     gemma_root=gemma_root,
     loras=[],
     quantization=QuantizationPolicy.fp8_cast(),  # keep FP8 quantization unchanged
@@ -522,6 +525,7 @@ _orig_video_decoder_factory = ledger.video_decoder
 _orig_audio_encoder_factory = ledger.audio_encoder
 _orig_audio_decoder_factory = ledger.audio_decoder
 _orig_vocoder_factory = ledger.vocoder
+_orig_spatial_upsampler_factory = ledger.spatial_upsampler
 _orig_text_encoder_factory = ledger.text_encoder
 _orig_gemma_embeddings_factory = ledger.gemma_embeddings_processor
 
@@ -532,6 +536,7 @@ _video_decoder = _orig_video_decoder_factory()
 _audio_encoder = _orig_audio_encoder_factory()
 _audio_decoder = _orig_audio_decoder_factory()
 _vocoder = _orig_vocoder_factory()
+_spatial_upsampler = _orig_spatial_upsampler_factory()
 _text_encoder = _orig_text_encoder_factory()
 _embeddings_processor = _orig_gemma_embeddings_factory()
 
@@ -543,6 +548,7 @@ ledger.video_decoder = lambda: _video_decoder
 ledger.audio_encoder = lambda: _audio_encoder
 ledger.audio_decoder = lambda: _audio_decoder
 ledger.vocoder = lambda: _vocoder
+ledger.spatial_upsampler = lambda: _spatial_upsampler
 ledger.text_encoder = lambda: _text_encoder
 ledger.gemma_embeddings_processor = lambda: _embeddings_processor