Update handler.py

handler.py

@@ -147,7 +147,7 @@ class EndpointHandler:
         #torch.backends.cuda.matmul.allow_tf32 = True
 
         # use distilled weights
-        model_path = f"{model_path}/ltxv-2b-0.9.6-distilled-04-25.safetensors"
+        model_path = "/repository/ltxv-2b-0.9.6-distilled-04-25.safetensors"
 
         transformer = LTXVideoTransformer3DModel.from_single_file(
             model_path, torch_dtype=torch.bfloat16
@@ -157,7 +157,7 @@ class EndpointHandler:
 
         if support_image_prompt:
             self.image_to_video = LTXImageToVideoPipeline.from_pretrained(
-                model_path,
+                "/repository",
                 transformer=transformer,
                 vae=vae,
                 torch_dtype=torch.bfloat16
@@ -175,7 +175,7 @@ class EndpointHandler:
         else:
             # Initialize models with bfloat16 precision
             self.text_to_video = LTXPipeline.from_pretrained(
-                model_path,
+                "/repository",
                 transformer=transformer,
                 vae=vae,
                 torch_dtype=torch.bfloat16