Update api/ltx/ltx_utils.py

api/ltx/ltx_utils.py

@@ -75,17 +75,22 @@ def build_ltx_pipeline_on_cpu(config: Dict) -> Tuple[LTXVideoPipeline, Optional[
     t0 = time.perf_counter()
     logging.info("Building LTX pipeline on CPU...")
 
-    ckpt_path = hf_hub_download(repo_id=LTX_REPO_ID, filename=self.config["checkpoint_path"], cache_dir=CACHE_DIR)
-        
+
+
+    ckpt_path = hf_hub_download(repo_id=LTX_REPO_ID, filename=config["checkpoint_path"], cache_dir=CACHE_DIR)
+   
     if not ckpt_path.is_file():
         raise FileNotFoundError(f"Main checkpoint file not found: {ckpt_path}")
+        
 
+    
     with safe_open(ckpt_path, framework="pt") as f:
         metadata = f.metadata() or {}
         config_str = metadata.get("config", "{}")
         configs = json.loads(config_str)
         allowed_inference_steps = configs.get("allowed_inference_steps")
 
+        
     vae = CausalVideoAutoencoder.from_pretrained(ckpt_path).to("cpu")
     transformer = Transformer3DModel.from_pretrained(ckpt_path).to("cpu")
     scheduler = RectifiedFlowScheduler.from_pretrained(ckpt_path)