Spaces:
Paused
Paused
Update api/ltx/ltx_aduc_pipeline.py
Browse files
api/ltx/ltx_aduc_pipeline.py
CHANGED
|
@@ -179,9 +179,10 @@ class LtxAducPipeline:
|
|
| 179 |
t0 = time.perf_counter()
|
| 180 |
logging.info("Carregando sub-modelos do LTX para a CPU...")
|
| 181 |
|
| 182 |
-
|
|
|
|
| 183 |
if not ckpt_path.is_file():
|
| 184 |
-
raise FileNotFoundError(f"
|
| 185 |
|
| 186 |
# 1. Carrega Metadados do Checkpoint
|
| 187 |
with safe_open(ckpt_path, framework="pt") as f:
|
|
|
|
| 179 |
t0 = time.perf_counter()
|
| 180 |
logging.info("Carregando sub-modelos do LTX para a CPU...")
|
| 181 |
|
| 182 |
+
ckpt_path_str = hf_hub_download(repo_id=LTX_REPO_ID, filename=config["checkpoint_path"], cache_dir=CACHE_DIR)
|
| 183 |
+
ckpt_path = Path(ckpt_path_str)
|
| 184 |
if not ckpt_path.is_file():
|
| 185 |
+
raise FileNotFoundError(f"Main checkpoint file not found: {ckpt_path}")
|
| 186 |
|
| 187 |
# 1. Carrega Metadados do Checkpoint
|
| 188 |
with safe_open(ckpt_path, framework="pt") as f:
|