Spaces:
Paused
Paused
Update api/ltx_server.py
Browse files- api/ltx_server.py +1 -1
api/ltx_server.py
CHANGED
|
@@ -304,7 +304,6 @@ class VideoService:
|
|
| 304 |
local_dir=os.getenv("HF_HOME"),
|
| 305 |
cache_dir=os.getenv("HF_HOME_CACHE"),
|
| 306 |
token=os.getenv("HF_TOKEN"),
|
| 307 |
-
safety_checker=None,
|
| 308 |
)
|
| 309 |
self.config["checkpoint_path"] = distilled_model_path
|
| 310 |
print(f"[DEBUG] Checkpoint em: {distilled_model_path}")
|
|
@@ -330,6 +329,7 @@ class VideoService:
|
|
| 330 |
enhance_prompt=False,
|
| 331 |
prompt_enhancer_image_caption_model_name_or_path=self.config["prompt_enhancer_image_caption_model_name_or_path"],
|
| 332 |
prompt_enhancer_llm_model_name_or_path=self.config["prompt_enhancer_llm_model_name_or_path"],
|
|
|
|
| 333 |
)
|
| 334 |
print("[DEBUG] Pipeline pronto.")
|
| 335 |
|
|
|
|
| 304 |
local_dir=os.getenv("HF_HOME"),
|
| 305 |
cache_dir=os.getenv("HF_HOME_CACHE"),
|
| 306 |
token=os.getenv("HF_TOKEN"),
|
|
|
|
| 307 |
)
|
| 308 |
self.config["checkpoint_path"] = distilled_model_path
|
| 309 |
print(f"[DEBUG] Checkpoint em: {distilled_model_path}")
|
|
|
|
| 329 |
enhance_prompt=False,
|
| 330 |
prompt_enhancer_image_caption_model_name_or_path=self.config["prompt_enhancer_image_caption_model_name_or_path"],
|
| 331 |
prompt_enhancer_llm_model_name_or_path=self.config["prompt_enhancer_llm_model_name_or_path"],
|
| 332 |
+
safety_checker=None,
|
| 333 |
)
|
| 334 |
print("[DEBUG] Pipeline pronto.")
|
| 335 |
|