Update api/ltx_server.py

api/ltx_server.py

@@ -356,29 +356,27 @@ class VideoService:
     def _load_models(self):
         t0 = time.perf_counter()
         LTX_REPO = "Lightricks/LTX-Video"
-        #print("[DEBUG] Baixando checkpoint principal...")
-        #distilled_model_path = LTX_REPO / str(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}")
-
-        #print("[DEBUG] Baixando upscaler espacial...")
-        #spatial_upscaler_path = LTX_REPO / str(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("[DEBUG] Baixando checkpoint principal...")
+        distilled_model_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}")
+
+        print("[DEBUG] Baixando upscaler espacial...")
+        spatial_upscaler_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("[DEBUG] Construindo pipeline...")
         pipeline = create_ltx_video_pipeline(