Carlos s
commited on
Update api/ltx_server.py
Browse files- 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
|