Spaces:
Paused
Paused
Update api/ltx/ltx_aduc_pipeline.py
Browse files
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}")
|