EuuIia commited on
Commit
0eed3f3
·
verified ·
1 Parent(s): 597520d

Update api/ltx_server.py

Browse files
Files changed (1) hide show
  1. api/ltx_server.py +17 -16
api/ltx_server.py CHANGED
@@ -705,22 +705,23 @@ class VideoService:
705
  print(f"[DEBUG] Latentes (multi-escala): shape={tuple(latents.shape)}")
706
  else:
707
  single_pass_kwargs = call_kwargs.copy()
708
- first_pass_config = self.config.get("first_pass", {})
709
- single_pass_kwargs.update(
710
- {
711
- "guidance_scale": float(guidance_scale),
712
- "stg_scale": first_pass_config.get("stg_scale"),
713
- "rescaling_scale": first_pass_config.get("rescaling_scale"),
714
- "skip_block_list": first_pass_config.get("skip_block_list"),
715
- }
716
- )
717
- schedule = first_pass_config.get("timesteps") or first_pass_config.get("guidance_timesteps")
718
- if mode == "video-to-video":
719
- schedule = [0.7]; print("[INFO] Modo video-to-video (etapa única): timesteps=[0.7]")
720
- if isinstance(schedule, (list, tuple)) and len(schedule) > 0:
721
- single_pass_kwargs["timesteps"] = schedule
722
- single_pass_kwargs["guidance_timesteps"] = schedule
723
- print(f"[DEBUG] Single-pass: timesteps_len={len(schedule) if schedule else 0}")
 
724
 
725
  print("\n[INFO] Executando pipeline de etapa única...")
726
  t_sp = time.perf_counter()
 
705
  print(f"[DEBUG] Latentes (multi-escala): shape={tuple(latents.shape)}")
706
  else:
707
  single_pass_kwargs = call_kwargs.copy()
708
+ first_pass_config = self.config.get("first_pass", {}).copy()
709
+
710
+ #single_pass_kwargs.update(
711
+ # {
712
+ #"guidance_scale": float(guidance_scale),
713
+ #"stg_scale": first_pass_config.get("stg_scale"),
714
+ #"rescaling_scale": first_pass_config.get("rescaling_scale"),
715
+ #"skip_block_list": first_pass_config.get("skip_block_list"),
716
+ # }
717
+ #)
718
+ #schedule = first_pass_config.get("timesteps") or first_pass_config.get("guidance_timesteps")
719
+ #if mode == "video-to-video":
720
+ # schedule = [0.7]; print("[INFO] Modo video-to-video (etapa única): timesteps=[0.7]")
721
+ #if isinstance(schedule, (list, tuple)) and len(schedule) > 0:
722
+ # single_pass_kwargs["timesteps"] = schedule
723
+ # single_pass_kwargs["guidance_timesteps"] = schedule
724
+ #print(f"[DEBUG] Single-pass: timesteps_len={len(schedule) if schedule else 0}")
725
 
726
  print("\n[INFO] Executando pipeline de etapa única...")
727
  t_sp = time.perf_counter()