Carlos s commited on
Commit
525ef07
·
verified ·
1 Parent(s): 2921cee

Update api/ltx_server.py

Browse files
Files changed (1) hide show
  1. 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