Spaces:
Paused
Paused
Update api/ltx_server.py
Browse files- api/ltx_server.py +3 -3
api/ltx_server.py
CHANGED
|
@@ -574,7 +574,7 @@ class VideoService:
|
|
| 574 |
|
| 575 |
|
| 576 |
|
| 577 |
-
def
|
| 578 |
total = latents_brutos.shape[2] # dimensão temporal (número de latentes)
|
| 579 |
|
| 580 |
if total % 2 == 1: # ÍMPAR
|
|
@@ -865,7 +865,7 @@ class VideoService:
|
|
| 865 |
#latents_parts_up = self._dividir_latentes_por_tamanho(latents_cpu_up,15,0)
|
| 866 |
|
| 867 |
|
| 868 |
-
lat_aup, lat_bup =
|
| 869 |
print(f"[DEBUG] Partição Aup: {tuple(lat_aup.shape)}")
|
| 870 |
print(f"[DEBUG] Partição Bup: {tuple(lat_bup.shape)}")
|
| 871 |
|
|
@@ -946,7 +946,7 @@ class VideoService:
|
|
| 946 |
#latents_parts_vae = self._dividir_latentes_por_tamanho(latents_cpu_vae,4,1)
|
| 947 |
|
| 948 |
|
| 949 |
-
lat_a, lat_b =
|
| 950 |
print(f"[DEBUG] Partição A: {tuple(lat_a.shape)}")
|
| 951 |
print(f"[DEBUG] Partição B: {tuple(lat_b.shape)}")
|
| 952 |
|
|
|
|
| 574 |
|
| 575 |
|
| 576 |
|
| 577 |
+
def _dividir_latentes(self, latents_brutos):
|
| 578 |
total = latents_brutos.shape[2] # dimensão temporal (número de latentes)
|
| 579 |
|
| 580 |
if total % 2 == 1: # ÍMPAR
|
|
|
|
| 865 |
#latents_parts_up = self._dividir_latentes_por_tamanho(latents_cpu_up,15,0)
|
| 866 |
|
| 867 |
|
| 868 |
+
lat_aup, lat_bup = self._dividir_latentes(latents_parts_up)
|
| 869 |
print(f"[DEBUG] Partição Aup: {tuple(lat_aup.shape)}")
|
| 870 |
print(f"[DEBUG] Partição Bup: {tuple(lat_bup.shape)}")
|
| 871 |
|
|
|
|
| 946 |
#latents_parts_vae = self._dividir_latentes_por_tamanho(latents_cpu_vae,4,1)
|
| 947 |
|
| 948 |
|
| 949 |
+
lat_a, lat_b = self._dividir_latentes(latents_cpu_vae)
|
| 950 |
print(f"[DEBUG] Partição A: {tuple(lat_a.shape)}")
|
| 951 |
print(f"[DEBUG] Partição B: {tuple(lat_b.shape)}")
|
| 952 |
|