Update api/ltx/ltx_aduc_pipeline.py

api/ltx/ltx_aduc_pipeline.py

@@ -235,9 +235,11 @@ class LtxAducPipeline:
 
         # 5. Carrega o Latent Upsampler (também na CPU)
         latent_upsampler = None
-        if self.config.get("spatial_upscaler_model_path"):
-            logging.info("Carregando Latent Upsampler...")
-            spatial_path = self.config["spatial_upscaler_model_path"]
+        if config.get("spatial_upscaler_model_path"):  
+            spatial_path_str = hf_hub_download(repo_id=LTX_REPO_ID, filename=config["spatial_upscaler_model_path"], cache_dir=CACHE_DIR)
+            spatial_path = Path(spatial_path_str)
+            if not ckpt_path.is_file():
+                raise FileNotFoundError(f"Main checkpoint file not found: {spatial_path}")
             latent_upsampler = create_latent_upsampler(spatial_path, device="cpu")
             if precision == "bfloat16":
                 latent_upsampler.to(torch.bfloat16)