Update app.py

app.py

@@ -43,8 +43,8 @@ from huggingface_hub import hf_hub_download
 from ltx_core.model.video_vae import TilingConfig, get_video_chunks_number
 from ltx_core.quantization import QuantizationPolicy
 from ltx_core.text_encoders.gemma.embeddings_processor import EmbeddingsProcessorOutput
+import ltx_pipelines.distilled as distilled_module
 from ltx_pipelines.distilled import DistilledPipeline
-from ltx_pipelines.utils import helpers as pipeline_helpers
 from ltx_pipelines.utils.args import ImageConditioningInput
 from ltx_pipelines.utils.media_io import encode_video
 
@@ -124,7 +124,7 @@ print("Pipeline ready!")
 print("=" * 80)
 
 
-@spaces.GPU(duration=120, size='xlarge')
+@spaces.GPU(duration=300)
 def generate_video(
     input_image,
     prompt: str,
@@ -195,15 +195,15 @@ def generate_video(
                 f"Please ensure {TEXT_ENCODER_SPACE} is running properly."
             )
 
-        # Monkey-patch encode_prompts to return pre-computed embeddings
-        # instead of loading the text encoder + embeddings processor
+        # Monkey-patch encode_prompts on the distilled module directly
+        # (it imports encode_prompts from helpers, so we must patch the local reference)
         precomputed = EmbeddingsProcessorOutput(
             video_encoding=video_context,
             audio_encoding=audio_context,
             attention_mask=torch.ones(1, device="cuda"),  # dummy mask
         )
-        original_encode_prompts = pipeline_helpers.encode_prompts
-        pipeline_helpers.encode_prompts = lambda *args, **kwargs: [precomputed]
+        original_encode_prompts = distilled_module.encode_prompts
+        distilled_module.encode_prompts = lambda *args, **kwargs: [precomputed]
 
         try:
             tiling_config = TilingConfig.default()
@@ -233,7 +233,7 @@ def generate_video(
             return str(output_path), current_seed
         finally:
             # Restore original encode_prompts
-            pipeline_helpers.encode_prompts = original_encode_prompts
+            distilled_module.encode_prompts = original_encode_prompts
 
     except Exception as e:
         import traceback