Update api/ltx_server.py
Browse files- api/ltx_server.py +5 -5
api/ltx_server.py
CHANGED
|
@@ -357,7 +357,7 @@ class VideoService:
|
|
| 357 |
t0 = time.perf_counter()
|
| 358 |
LTX_REPO = "Lightricks/LTX-Video"
|
| 359 |
#print("[DEBUG] Baixando checkpoint principal...")
|
| 360 |
-
distilled_model_path = LTX_REPO / str(self.config["checkpoint_path"])
|
| 361 |
#hf_hub_download(
|
| 362 |
# repo_id=LTX_REPO,
|
| 363 |
# filename=self.config["checkpoint_path"],
|
|
@@ -365,11 +365,11 @@ class VideoService:
|
|
| 365 |
# cache_dir=os.getenv("HF_HOME_CACHE"),
|
| 366 |
# token=os.getenv("HF_TOKEN"),
|
| 367 |
#)
|
| 368 |
-
self.config["checkpoint_path"] = distilled_model_path
|
| 369 |
-
print(f"[DEBUG] Checkpoint em: {distilled_model_path}")
|
| 370 |
|
| 371 |
#print("[DEBUG] Baixando upscaler espacial...")
|
| 372 |
-
spatial_upscaler_path = LTX_REPO / str(self.config["spatial_upscaler_model_path"])
|
| 373 |
#hf_hub_download(
|
| 374 |
# repo_id=LTX_REPO,
|
| 375 |
# filename=self.config["spatial_upscaler_model_path"],
|
|
@@ -378,7 +378,7 @@ class VideoService:
|
|
| 378 |
# token=os.getenv("HF_TOKEN")
|
| 379 |
#)
|
| 380 |
#self.config["spatial_upscaler_model_path"] = spatial_upscaler_path
|
| 381 |
-
print(f"[DEBUG] Upscaler em: {spatial_upscaler_path}")
|
| 382 |
|
| 383 |
print("[DEBUG] Construindo pipeline...")
|
| 384 |
pipeline = create_ltx_video_pipeline(
|
|
|
|
| 357 |
t0 = time.perf_counter()
|
| 358 |
LTX_REPO = "Lightricks/LTX-Video"
|
| 359 |
#print("[DEBUG] Baixando checkpoint principal...")
|
| 360 |
+
#distilled_model_path = LTX_REPO / str(self.config["checkpoint_path"])
|
| 361 |
#hf_hub_download(
|
| 362 |
# repo_id=LTX_REPO,
|
| 363 |
# filename=self.config["checkpoint_path"],
|
|
|
|
| 365 |
# cache_dir=os.getenv("HF_HOME_CACHE"),
|
| 366 |
# token=os.getenv("HF_TOKEN"),
|
| 367 |
#)
|
| 368 |
+
#self.config["checkpoint_path"] = distilled_model_path
|
| 369 |
+
#print(f"[DEBUG] Checkpoint em: {distilled_model_path}")
|
| 370 |
|
| 371 |
#print("[DEBUG] Baixando upscaler espacial...")
|
| 372 |
+
#spatial_upscaler_path = LTX_REPO / str(self.config["spatial_upscaler_model_path"])
|
| 373 |
#hf_hub_download(
|
| 374 |
# repo_id=LTX_REPO,
|
| 375 |
# filename=self.config["spatial_upscaler_model_path"],
|
|
|
|
| 378 |
# token=os.getenv("HF_TOKEN")
|
| 379 |
#)
|
| 380 |
#self.config["spatial_upscaler_model_path"] = spatial_upscaler_path
|
| 381 |
+
#print(f"[DEBUG] Upscaler em: {spatial_upscaler_path}")
|
| 382 |
|
| 383 |
print("[DEBUG] Construindo pipeline...")
|
| 384 |
pipeline = create_ltx_video_pipeline(
|