Update api/ltx/ltx_utils.py

api/ltx/ltx_utils.py

@@ -109,8 +109,11 @@ def build_ltx_pipeline_on_cpu(config: Dict) -> Tuple[LTXVideoPipeline, Optional[
     )
 
     latent_upsampler = None
-    if config.get("spatial_upscaler_model_path"):
-        spatial_path = 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)