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

Update api/ltx/ltx_aduc_pipeline.py

Browse files
Files changed (1) hide show
  1. api/ltx/ltx_aduc_pipeline.py +2 -2
api/ltx/ltx_aduc_pipeline.py CHANGED
@@ -179,7 +179,7 @@ class LtxAducPipeline:
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}")
@@ -237,7 +237,7 @@ class LtxAducPipeline:
237
  # 5. Carrega o Latent Upsampler (também na CPU)
238
  latent_upsampler = None
239
  if config.get("spatial_upscaler_model_path"):
240
- spatial_path_str = hf_hub_download(repo_id=LTX_REPO_ID, filename=config["spatial_upscaler_model_path"], cache_dir=CACHE_DIR)
241
  spatial_path = Path(spatial_path_str)
242
  if not ckpt_path.is_file():
243
  raise FileNotFoundError(f"Main checkpoint file not found: {spatial_path}")
 
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=self.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}")
 
237
  # 5. Carrega o Latent Upsampler (também na CPU)
238
  latent_upsampler = None
239
  if config.get("spatial_upscaler_model_path"):
240
+ spatial_path_str = hf_hub_download(repo_id=LTX_REPO_ID, filename=self.config["spatial_upscaler_model_path"], cache_dir=CACHE_DIR)
241
  spatial_path = Path(spatial_path_str)
242
  if not ckpt_path.is_file():
243
  raise FileNotFoundError(f"Main checkpoint file not found: {spatial_path}")