Update api/ltx_server.py

api/ltx_server.py

@@ -355,28 +355,30 @@ class VideoService:
 
     def _load_models(self):
         t0 = time.perf_counter()
-        #LTX_REPO = "Lightricks/LTX-Video"
+        LTX_REPO = "Lightricks/LTX-Video"
         #print("[DEBUG] Baixando checkpoint principal...")
-        #distilled_model_path = hf_hub_download(
+        distilled_model_path = self.config["checkpoint_path"]
+        #hf_hub_download(
         #    repo_id=LTX_REPO,
         #    filename=self.config["checkpoint_path"],
         #    local_dir=os.getenv("HF_HOME"),
         #    cache_dir=os.getenv("HF_HOME_CACHE"),
         #    token=os.getenv("HF_TOKEN"),
         #)
-        #self.config["checkpoint_path"] = distilled_model_path
-        #print(f"[DEBUG] Checkpoint em: {distilled_model_path}")
+        self.config["checkpoint_path"] = distilled_model_path
+        print(f"[DEBUG] Checkpoint em: {distilled_model_path}")
 
         #print("[DEBUG] Baixando upscaler espacial...")
-        #spatial_upscaler_path = hf_hub_download(
+        spatial_upscaler_path = self.config["spatial_upscaler_model_path"],
+        hf_hub_download(
         #    repo_id=LTX_REPO,
         #    filename=self.config["spatial_upscaler_model_path"],
         #    local_dir=os.getenv("HF_HOME"),
         #    cache_dir=os.getenv("HF_HOME_CACHE"),
         #    token=os.getenv("HF_TOKEN")
-        )
+        #)
         #self.config["spatial_upscaler_model_path"] = spatial_upscaler_path
-        #print(f"[DEBUG] Upscaler em: {spatial_upscaler_path}")
+        print(f"[DEBUG] Upscaler em: {spatial_upscaler_path}")
 
         print("[DEBUG] Construindo pipeline...")
         pipeline = create_ltx_video_pipeline(