Spaces:
Paused
Paused
Update api/ltx/ltx_utils.py
Browse files- 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 |
-
|
| 81 |
-
|
| 82 |
-
|
| 83 |
-
|
| 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 |
|