Carlos s
commited on
Update api/ltx_server.py
Browse files- api/ltx_server.py +6 -1
api/ltx_server.py
CHANGED
|
@@ -620,11 +620,16 @@ class VideoService:
|
|
| 620 |
# =======================================================================
|
| 621 |
|
| 622 |
print(f"[DEBUG] Passo 1: Dimensões reduzidas e ajustadas para {first_pass_kwargs['height']}x{first_pass_kwargs['width']}")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 623 |
|
| 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
|
| 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}")
|
|
|
|
| 620 |
# =======================================================================
|
| 621 |
|
| 622 |
print(f"[DEBUG] Passo 1: Dimensões reduzidas e ajustadas para {first_pass_kwargs['height']}x{first_pass_kwargs['width']}")
|
| 623 |
+
|
| 624 |
+
|
| 625 |
+
first_pass_kwargs["output_type"] = "latent"
|
| 626 |
+
|
| 627 |
+
print(f"[DEBUG][first_pass_kwargs] {first_pass_kwargs}")
|
| 628 |
|
| 629 |
t_p1 = time.perf_counter()
|
| 630 |
ctx = torch.autocast(device_type="cuda", dtype=self.runtime_autocast_dtype) if self.device == "cuda" else contextlib.nullcontext()
|
| 631 |
with ctx:
|
| 632 |
+
first_pass_result = self.pipeline(**first_pass_kwargs)
|
| 633 |
|
| 634 |
latents_low_res = first_pass_result.latents if hasattr(first_pass_result, "latents") else first_pass_result
|
| 635 |
print(f"[DEBUG] Passo 1 concluído. Shape dos latentes de baixa resolução: {latents_low_res.shape}")
|