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

Update api/ltx_server.py

Browse files
Files changed (1) hide show
  1. api/ltx_server.py +2 -1
api/ltx_server.py CHANGED
@@ -594,6 +594,7 @@ 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
  })
598
  schedule = first_pass_args.get("timesteps") or first_pass_args.get("guidance_timesteps")
599
  if schedule:
@@ -627,7 +628,7 @@ class VideoService:
627
  first_pass_result = self.pipeline(**first_pass_kwargs)
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 em {time.perf_counter()-t_p1:.3f}s. Shape dos latentes de baixa resolução: {tuple(latents_low_res.shape)}")
631
  log_tensor_info(latents_low_res, "Latentes (Passo 1)")
632
 
633
  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
+ "output_type": "latent",
598
  })
599
  schedule = first_pass_args.get("timesteps") or first_pass_args.get("guidance_timesteps")
600
  if schedule:
 
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