eeuuia commited on
Commit
2f2fce0
·
verified ·
1 Parent(s): f2755cf

Update api/ltx/ltx_aduc_pipeline.py

Browse files
Files changed (1) hide show
  1. api/ltx/ltx_aduc_pipeline.py +3 -2
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
- ckpt_path = Path(self.config["checkpoint_path"])
 
183
  if not ckpt_path.is_file():
184
- raise FileNotFoundError(f"Arquivo de checkpoint principal não encontrado em: {ckpt_path}")
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: