Carlos s commited on
Commit
d7dea55
·
verified ·
1 Parent(s): c25c67b

Update api/ltx_server.py

Browse files
Files changed (1) hide show
  1. api/ltx_server.py +4 -0
api/ltx_server.py CHANGED
@@ -622,6 +622,10 @@ class VideoService:
622
  latents = result_first
623
  print(f"[DEBUG] Latentes (single-pass): shape={tuple(latents.shape)}")
624
  latents_first = latents
 
 
 
 
625
 
626
  # --- SECOND PASS ---
627
  print("[DEBUG] Executando SECOND PASS (latent_upsampler)...")
 
622
  latents = result_first
623
  print(f"[DEBUG] Latentes (single-pass): shape={tuple(latents.shape)}")
624
  latents_first = latents
625
+
626
+
627
+ ctx = torch.autocast(device_type="cuda", dtype=self.runtime_autocast_dtype) if self.device == "cuda" else contextlib.nullcontext()
628
+
629
 
630
  # --- SECOND PASS ---
631
  print("[DEBUG] Executando SECOND PASS (latent_upsampler)...")