Carlos s commited on
Commit
9e2b895
·
verified ·
1 Parent(s): 146467d

Update api/ltx_server.py

Browse files
Files changed (1) hide show
  1. api/ltx_server.py +6 -1
api/ltx_server.py CHANGED
@@ -620,11 +620,16 @@ class VideoService:
620
  # =======================================================================
621
 
622
  print(f"[DEBUG] Passo 1: Dimensões reduzidas e ajustadas para {first_pass_kwargs['height']}x{first_pass_kwargs['width']}")
 
 
 
 
 
623
 
624
  t_p1 = time.perf_counter()
625
  ctx = torch.autocast(device_type="cuda", dtype=self.runtime_autocast_dtype) if self.device == "cuda" else contextlib.nullcontext()
626
  with ctx:
627
- first_pass_result = self.pipeline(**first_pass_kwargs, "output_type": "latent")
628
 
629
  latents_low_res = first_pass_result.latents if hasattr(first_pass_result, "latents") else first_pass_result
630
  print(f"[DEBUG] Passo 1 concluído. Shape dos latentes de baixa resolução: {latents_low_res.shape}")
 
620
  # =======================================================================
621
 
622
  print(f"[DEBUG] Passo 1: Dimensões reduzidas e ajustadas para {first_pass_kwargs['height']}x{first_pass_kwargs['width']}")
623
+
624
+
625
+ first_pass_kwargs["output_type"] = "latent"
626
+
627
+ print(f"[DEBUG][first_pass_kwargs] {first_pass_kwargs}")
628
 
629
  t_p1 = time.perf_counter()
630
  ctx = torch.autocast(device_type="cuda", dtype=self.runtime_autocast_dtype) if self.device == "cuda" else contextlib.nullcontext()
631
  with ctx:
632
+ first_pass_result = self.pipeline(**first_pass_kwargs)
633
 
634
  latents_low_res = first_pass_result.latents if hasattr(first_pass_result, "latents") else first_pass_result
635
  print(f"[DEBUG] Passo 1 concluído. Shape dos latentes de baixa resolução: {latents_low_res.shape}")