EuuIia commited on
Commit
bb90730
·
verified ·
1 Parent(s): 6077505

Update api/ltx_server.py

Browse files
Files changed (1) hide show
  1. api/ltx_server.py +5 -3
api/ltx_server.py CHANGED
@@ -620,9 +620,11 @@ class VideoService:
620
  ctx = torch.autocast(device_type="cuda", dtype=self.runtime_autocast_dtype) if self.device == "cuda" else contextlib.nullcontext()
621
  with ctx:
622
  first_pass_result = self.pipeline(**first_pass_kwargs)
623
-
624
- latents_low_res = first_pass_result.latents if hasattr(first_pass_result, "latents") else first_pass_result
625
- log_tensor_info(latents_low_res, "Latentes (Passo 1)")
 
 
626
 
627
  del first_pass_result
628
  gc.collect()
 
620
  ctx = torch.autocast(device_type="cuda", dtype=self.runtime_autocast_dtype) if self.device == "cuda" else contextlib.nullcontext()
621
  with ctx:
622
  first_pass_result = self.pipeline(**first_pass_kwargs)
623
+
624
+
625
+ latents_low_res = first_pass_result.images
626
+ #latents_low_res = first_pass_result.latents if hasattr(first_pass_result, "latents") else first_pass_result
627
+ log_tensor_info(latents_low_res, f"Latentes (Passo 1) {latents_low_res}")
628
 
629
  del first_pass_result
630
  gc.collect()