EuuIia commited on
Commit
872c7c8
·
verified ·
1 Parent(s): c2ee5af

Update api/ltx_server.py

Browse files
Files changed (1) hide show
  1. 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 dividir_latentes(latents_brutos):
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 = 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
 
 
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