Spaces:
Paused
Paused
Update api/ltx_server.py
Browse files- api/ltx_server.py +2 -2
api/ltx_server.py
CHANGED
|
@@ -396,7 +396,7 @@ class VideoService:
|
|
| 396 |
print(f"[DEBUG] Cond shape={tuple(out.shape)} dtype={out.dtype} device={out.device}")
|
| 397 |
return out
|
| 398 |
|
| 399 |
-
def
|
| 400 |
total = latents_brutos.shape[2] # dimensão temporal (número de latentes)
|
| 401 |
|
| 402 |
if total % 2 == 1: # ÍMPAR
|
|
@@ -609,7 +609,7 @@ class VideoService:
|
|
| 609 |
pass
|
| 610 |
|
| 611 |
# 2) Divide em duas partes
|
| 612 |
-
lat_a, lat_b =
|
| 613 |
print(f"[DEBUG] Partição A: {tuple(lat_a.shape)}")
|
| 614 |
print(f"[DEBUG] Partição B: {tuple(lat_b.shape)}")
|
| 615 |
|
|
|
|
| 396 |
print(f"[DEBUG] Cond shape={tuple(out.shape)} dtype={out.dtype} device={out.device}")
|
| 397 |
return out
|
| 398 |
|
| 399 |
+
def _dividir_latentes(self, latents_brutos):
|
| 400 |
total = latents_brutos.shape[2] # dimensão temporal (número de latentes)
|
| 401 |
|
| 402 |
if total % 2 == 1: # ÍMPAR
|
|
|
|
| 609 |
pass
|
| 610 |
|
| 611 |
# 2) Divide em duas partes
|
| 612 |
+
lat_a, lat_b = self._dividir_latentes(latents_cpu)
|
| 613 |
print(f"[DEBUG] Partição A: {tuple(lat_a.shape)}")
|
| 614 |
print(f"[DEBUG] Partição B: {tuple(lat_b.shape)}")
|
| 615 |
|