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

Update api/ltx_server.py

Browse files
Files changed (1) hide show
  1. api/ltx_server.py +2 -3
api/ltx_server.py CHANGED
@@ -594,7 +594,6 @@ class VideoService:
594
  "stg_scale": first_pass_args.get("stg_scale"),
595
  "rescaling_scale": first_pass_args.get("rescaling_scale"),
596
  "skip_block_list": first_pass_args.get("skip_block_list"),
597
- "output_type": "latent",
598
  })
599
  schedule = first_pass_args.get("timesteps") or first_pass_args.get("guidance_timesteps")
600
  if schedule:
@@ -625,10 +624,10 @@ class VideoService:
625
  t_p1 = time.perf_counter()
626
  ctx = torch.autocast(device_type="cuda", dtype=self.runtime_autocast_dtype) if self.device == "cuda" else contextlib.nullcontext()
627
  with ctx:
628
- first_pass_result = self.pipeline(**first_pass_kwargs)
629
 
630
  latents_low_res = first_pass_result.latents if hasattr(first_pass_result, "latents") else first_pass_result
631
- print(f"[DEBUG] Passo 1 concluído. Shape dos latentes de baixa resolução: {latents_low_res}")
632
  log_tensor_info(latents_low_res, "Latentes (Passo 1)")
633
 
634
  del first_pass_result, first_pass_kwargs
 
594
  "stg_scale": first_pass_args.get("stg_scale"),
595
  "rescaling_scale": first_pass_args.get("rescaling_scale"),
596
  "skip_block_list": first_pass_args.get("skip_block_list"),
 
597
  })
598
  schedule = first_pass_args.get("timesteps") or first_pass_args.get("guidance_timesteps")
599
  if schedule:
 
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}")
631
  log_tensor_info(latents_low_res, "Latentes (Passo 1)")
632
 
633
  del first_pass_result, first_pass_kwargs