Spaces:
Paused
Paused
Update video_service.py
Browse files- 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 |
|