Update api/ltx_server.py
Browse files- api/ltx_server.py +5 -3
api/ltx_server.py
CHANGED
|
@@ -620,9 +620,11 @@ class VideoService:
|
|
| 620 |
ctx = torch.autocast(device_type="cuda", dtype=self.runtime_autocast_dtype) if self.device == "cuda" else contextlib.nullcontext()
|
| 621 |
with ctx:
|
| 622 |
first_pass_result = self.pipeline(**first_pass_kwargs)
|
| 623 |
-
|
| 624 |
-
|
| 625 |
-
|
|
|
|
|
|
|
| 626 |
|
| 627 |
del first_pass_result
|
| 628 |
gc.collect()
|
|
|
|
| 620 |
ctx = torch.autocast(device_type="cuda", dtype=self.runtime_autocast_dtype) if self.device == "cuda" else contextlib.nullcontext()
|
| 621 |
with ctx:
|
| 622 |
first_pass_result = self.pipeline(**first_pass_kwargs)
|
| 623 |
+
|
| 624 |
+
|
| 625 |
+
latents_low_res = first_pass_result.images
|
| 626 |
+
#latents_low_res = first_pass_result.latents if hasattr(first_pass_result, "latents") else first_pass_result
|
| 627 |
+
log_tensor_info(latents_low_res, f"Latentes (Passo 1) {latents_low_res}")
|
| 628 |
|
| 629 |
del first_pass_result
|
| 630 |
gc.collect()
|