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