EuuIia commited on
Commit
c14605c
·
verified ·
1 Parent(s): d612da0

Update video_service.py

Browse files
Files changed (1) hide show
  1. video_service.py +1 -1
video_service.py CHANGED
@@ -252,6 +252,7 @@ class VideoService:
252
  return yaml.safe_load(file)
253
 
254
  def _load_models(self):
 
255
  LTX_REPO = "Lightricks/LTX-Video"
256
  distilled_model_path = hf_hub_download(
257
  repo_id=LTX_REPO,
@@ -259,7 +260,6 @@ class VideoService:
259
  local_dir=os.getenv("HF_HOME"),
260
  cache_dir=os.getenv("HF_HOME_CACHE"),
261
  token=os.getenv("HF_TOKEN"),
262
- torch_dtype="Float8_e4m3fn",
263
  )
264
  self.config["checkpoint_path"] = distilled_model_path
265
 
 
252
  return yaml.safe_load(file)
253
 
254
  def _load_models(self):
255
+
256
  LTX_REPO = "Lightricks/LTX-Video"
257
  distilled_model_path = hf_hub_download(
258
  repo_id=LTX_REPO,
 
260
  local_dir=os.getenv("HF_HOME"),
261
  cache_dir=os.getenv("HF_HOME_CACHE"),
262
  token=os.getenv("HF_TOKEN"),
 
263
  )
264
  self.config["checkpoint_path"] = distilled_model_path
265