EuuIia commited on
Commit
a600797
·
verified ·
1 Parent(s): cf601a2

Update api/ltx_server.py

Browse files
Files changed (1) hide show
  1. api/ltx_server.py +7 -4
api/ltx_server.py CHANGED
@@ -644,11 +644,12 @@ class VideoService:
644
  pass
645
 
646
  # 2) Divide em duas partes
647
- lat_a, lat_b = self._dividir_latentes_em_partes(latents_cpu, 2)
648
- print(f"[DEBUG] Partição A: {tuple(lat_a.shape)}")
649
- print(f"[DEBUG] Partição B: {tuple(lat_b.shape)}")
 
 
650
 
651
- latents_parts = [lat_a, lat_b]
652
 
653
  temp_dir = tempfile.mkdtemp(prefix="ltxv_"); self._register_tmp_dir(temp_dir)
654
  results_dir = "/app/output"; os.makedirs(results_dir, exist_ok=True)
@@ -657,6 +658,8 @@ class VideoService:
657
  par = 0
658
 
659
  for latents in latents_parts:
 
 
660
  par = par + 1
661
  output_video_path = os.path.join(temp_dir, f"output_{used_seed}_{par}.mp4")
662
  final_output_path = None
 
644
  pass
645
 
646
  # 2) Divide em duas partes
647
+ #lat_a, lat_b = self._dividir_latentes_em_partes(latents_cpu, 2)
648
+ latents_parts = self._dividir_latentes_em_partes(latents_cpu, 2)
649
+
650
+ print(f"[DEBUG] Partição B: {tuple(lat_b.shape)}")
651
+
652
 
 
653
 
654
  temp_dir = tempfile.mkdtemp(prefix="ltxv_"); self._register_tmp_dir(temp_dir)
655
  results_dir = "/app/output"; os.makedirs(results_dir, exist_ok=True)
 
658
  par = 0
659
 
660
  for latents in latents_parts:
661
+ print(f"[DEBUG] Partição {par}: {tuple(latents.shape)}")
662
+
663
  par = par + 1
664
  output_video_path = os.path.join(temp_dir, f"output_{used_seed}_{par}.mp4")
665
  final_output_path = None