Carlos s
commited on
Update api/ltx_server.py
Browse files- api/ltx_server.py +1 -1
api/ltx_server.py
CHANGED
|
@@ -590,7 +590,7 @@ class VideoService:
|
|
| 590 |
first_pass_kwargs["width"] = width_p1
|
| 591 |
|
| 592 |
print(f"[DEBUG] Passo 1: Dimensões reduzidas e ajustadas para {height_p1}x{width_p1}")
|
| 593 |
-
|
| 594 |
with ctx:
|
| 595 |
first_pass_result = self.pipeline(**first_pass_kwargs)
|
| 596 |
|
|
|
|
| 590 |
first_pass_kwargs["width"] = width_p1
|
| 591 |
|
| 592 |
print(f"[DEBUG] Passo 1: Dimensões reduzidas e ajustadas para {height_p1}x{width_p1}")
|
| 593 |
+
ctx = torch.autocast(device_type="cuda", dtype=self.runtime_autocast_dtype) if self.device == "cuda" else contextlib.nullcontext()
|
| 594 |
with ctx:
|
| 595 |
first_pass_result = self.pipeline(**first_pass_kwargs)
|
| 596 |
|