Update api/ltx_server.py
Browse files- 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}")
|