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

Update api/ltx_server.py

Browse files
Files changed (1) hide show
  1. api/ltx_server.py +2 -1
api/ltx_server.py CHANGED
@@ -304,6 +304,7 @@ class VideoService:
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}")
@@ -314,7 +315,7 @@ class VideoService:
314
  filename=self.config["spatial_upscaler_model_path"],
315
  local_dir=os.getenv("HF_HOME"),
316
  cache_dir=os.getenv("HF_HOME_CACHE"),
317
- token=os.getenv("HF_TOKEN"),
318
  )
319
  self.config["spatial_upscaler_model_path"] = spatial_upscaler_path
320
  print(f"[DEBUG] Upscaler em: {spatial_upscaler_path}")
 
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}")
 
315
  filename=self.config["spatial_upscaler_model_path"],
316
  local_dir=os.getenv("HF_HOME"),
317
  cache_dir=os.getenv("HF_HOME_CACHE"),
318
+ token=os.getenv("HF_TOKEN")
319
  )
320
  self.config["spatial_upscaler_model_path"] = spatial_upscaler_path
321
  print(f"[DEBUG] Upscaler em: {spatial_upscaler_path}")