Update handler.py

handler.py

@@ -7,12 +7,12 @@ import io
 import logging
 import random
 import traceback
-
+import os
 import numpy as np
 import torch
 from diffusers import LTXPipeline, LTXImageToVideoPipeline
 from PIL import Image
-from varnish import Varnish
+from varnish import Varnish, setup_debug_logging
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
@@ -85,13 +85,17 @@ class EndpointHandler:
         #self.text_to_video.enable_model_cpu_offload()
         #self.image_to_video.enable_model_cpu_offload()
 
+        # temporary enable this if you have some issues with locating the model files
+        setup_debug_logging()
+
         # Initialize Varnish for post-processing
         self.varnish = Varnish(
             device="cuda" if torch.cuda.is_available() else "cpu",
             output_format="mp4",
             output_codec="h264",
             output_quality=23,
-            enable_mmaudio=False
+            enable_mmaudio=False,
+            #model_base_dir=os.path.abspath(os.path.join(os.getcwd(), "varnish"))
         )
 
     async def process_frames(