Spaces:
Paused
Paused
Update api/ltx_server.py
Browse files- api/ltx_server.py +1 -1
api/ltx_server.py
CHANGED
|
@@ -873,7 +873,7 @@ class VideoService:
|
|
| 873 |
latents = None
|
| 874 |
|
| 875 |
try:
|
| 876 |
-
ctx = torch.
|
| 877 |
|
| 878 |
if improve_texture:
|
| 879 |
if not self.latent_upsampler:
|
|
|
|
| 873 |
latents = None
|
| 874 |
|
| 875 |
try:
|
| 876 |
+
ctx = torch.autocast(device_type="cuda", dtype=self.runtime_autocast_dtype) if self.device == "cuda" else contextlib.nullcontext()
|
| 877 |
|
| 878 |
if improve_texture:
|
| 879 |
if not self.latent_upsampler:
|