Carlos s
commited on
Update api/ltx_server.py
Browse files- api/ltx_server.py +2 -2
api/ltx_server.py
CHANGED
|
@@ -398,7 +398,7 @@ class VideoService:
|
|
| 398 |
return out
|
| 399 |
|
| 400 |
|
| 401 |
-
def
|
| 402 |
# latents: [B, C, T, H, W]
|
| 403 |
T = latents.shape[2]
|
| 404 |
L = num_por_chunk + overlap # janela com stop exclusivo
|
|
@@ -743,7 +743,7 @@ class VideoService:
|
|
| 743 |
except Exception:
|
| 744 |
pass
|
| 745 |
|
| 746 |
-
latents_parts = self.
|
| 747 |
|
| 748 |
temp_dir = tempfile.mkdtemp(prefix="ltxv_"); self._register_tmp_dir(temp_dir)
|
| 749 |
results_dir = "/app/output"; os.makedirs(results_dir, exist_ok=True)
|
|
|
|
| 398 |
return out
|
| 399 |
|
| 400 |
|
| 401 |
+
def _dividir_latentes(self, latents, num_por_chunk: int, overlap: int = 1):
|
| 402 |
# latents: [B, C, T, H, W]
|
| 403 |
T = latents.shape[2]
|
| 404 |
L = num_por_chunk + overlap # janela com stop exclusivo
|
|
|
|
| 743 |
except Exception:
|
| 744 |
pass
|
| 745 |
|
| 746 |
+
latents_parts = self._dividir_latentes(latents_cpu,4,1)
|
| 747 |
|
| 748 |
temp_dir = tempfile.mkdtemp(prefix="ltxv_"); self._register_tmp_dir(temp_dir)
|
| 749 |
results_dir = "/app/output"; os.makedirs(results_dir, exist_ok=True)
|