EuuIia commited on
Commit
bd157db
·
verified ·
1 Parent(s): 58938aa

Update api/ltx_server.py

Browse files
Files changed (1) hide show
  1. 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