Eueuiaa commited on
Commit
9dd6c0e
·
verified ·
1 Parent(s): 8677c3e

Update api/ltx_server.py

Browse files
Files changed (1) hide show
  1. api/ltx_server.py +3 -3
api/ltx_server.py CHANGED
@@ -574,7 +574,7 @@ class VideoService:
574
 
575
 
576
 
577
- def dividir_latentes(latents_brutos):
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 = 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,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 = 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
 
 
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