eeuuia commited on
Commit
ece26e1
·
verified ·
1 Parent(s): b5a0623

Update api/ltx/ltx_utils.py

Browse files
Files changed (1) hide show
  1. api/ltx/ltx_utils.py +4 -4
api/ltx/ltx_utils.py CHANGED
@@ -77,10 +77,10 @@ def build_ltx_pipeline_on_cpu(config: Dict) -> Tuple[LTXVideoPipeline, Optional[
77
 
78
 
79
 
80
- ckpt_path = hf_hub_download(repo_id=LTX_REPO_ID, filename=config["checkpoint_path"], cache_dir=CACHE_DIR)
81
-
82
- #if not ckpt_path.is_file():
83
- # raise FileNotFoundError(f"Main checkpoint file not found: {ckpt_path}")
84
 
85
 
86
 
 
77
 
78
 
79
 
80
+ ckpt_path_str = hf_hub_download(repo_id=LTX_REPO_ID, filename=config["checkpoint_path"], cache_dir=CACHE_DIR)
81
+ ckpt_path = Path(ckpt_path_str)
82
+ if not ckpt_path.is_file():
83
+ raise FileNotFoundError(f"Main checkpoint file not found: {ckpt_path}")
84
 
85
 
86