Carlos s
commited on
Update api/ltx_server.py
Browse files- api/ltx_server.py +8 -2
api/ltx_server.py
CHANGED
|
@@ -611,7 +611,7 @@ class VideoService:
|
|
| 611 |
|
| 612 |
print("[DEBUG] Executando FIRST PASS (pipeline base)...")
|
| 613 |
with ctx:
|
| 614 |
-
result_first = self.pipeline(**first_pass_kwargs)
|
| 615 |
|
| 616 |
|
| 617 |
if hasattr(result_first, "latents"):
|
|
@@ -627,8 +627,14 @@ class VideoService:
|
|
| 627 |
print("[DEBUG] Executando SECOND PASS (latent_upsampler)...")
|
| 628 |
with ctx:
|
| 629 |
result = self.latent_upsampler(
|
|
|
|
| 630 |
latents=result_first,
|
| 631 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 632 |
)
|
| 633 |
|
| 634 |
#with ctx:
|
|
|
|
| 611 |
|
| 612 |
print("[DEBUG] Executando FIRST PASS (pipeline base)...")
|
| 613 |
with ctx:
|
| 614 |
+
result_first = self.pipeline(**first_pass_kwargs).frames
|
| 615 |
|
| 616 |
|
| 617 |
if hasattr(result_first, "latents"):
|
|
|
|
| 627 |
print("[DEBUG] Executando SECOND PASS (latent_upsampler)...")
|
| 628 |
with ctx:
|
| 629 |
result = self.latent_upsampler(
|
| 630 |
+
**second_pass_kwargs,
|
| 631 |
latents=result_first,
|
| 632 |
+
denoise_strength=0.4,
|
| 633 |
+
num_inference_steps=10,
|
| 634 |
+
decode_timestep=0.05,
|
| 635 |
+
image_cond_noise_scale=0.025,
|
| 636 |
+
generator=torch.Generator().manual_seed(0),
|
| 637 |
+
output_type="latent",
|
| 638 |
)
|
| 639 |
|
| 640 |
#with ctx:
|