Update api/ltx_server.py

api/ltx_server.py

@@ -421,18 +421,18 @@ class VideoService:
         print("================PODA CAUSAL=================")
         print(f"[DEBUG] TOTAL LATENTES = {sum_latent}")
         print(f"[DEBUG] Num LATENTES por chunk = {num_latente_por_chunk}")
-        print(f"[DEBUG] Número de chunks = {steps}")
+        print(f"[DEBUG] Número de chunks = {n_chunks}")
 
 
-        if steps > 1:
+        if n_chunks > 1:
             i = 0
             end=1
-            while i < steps:
+            while i < n_chunks:
                 start = end-1
                 end += num_latente_por_chunk
                 if end+2>=sum_latent:
                     end = sum_latent-1
-                    i = steps
+                    i = n_chunks
                 else:
                     i += 1
                 chunk = latents_brutos[:, :, start:end, :, :].clone()
@@ -752,7 +752,7 @@ class VideoService:
             except Exception:
                 pass
                 
-            latents_parts = self._dividir_latentes_por_tamanho(latents_cpu,5,1)
+            latents_parts = self._dividir_latentes_por_tamanho(latents_cpu,4,1)
             
             temp_dir = tempfile.mkdtemp(prefix="ltxv_"); self._register_tmp_dir(temp_dir)
             results_dir = "/app/output"; os.makedirs(results_dir, exist_ok=True)
@@ -794,10 +794,12 @@ class VideoService:
                     final_output_path = output_video_path
                     print(f"[DEBUG] Falha no move; usando tmp como final: {e}")
 
-            final_concat = os.path.join(results_dir, f"concat_fim_{used_seed}.mp4")
-            partes_mp4_fade = self._gerar_lista_com_transicoes(pasta=results_dir, video_paths=partes_mp4, crossfade_frames=8)
-            self._concat_mp4s_no_reencode(partes_mp4_fade, final_concat)
-            
+            if (range(partes_mp4)>1)
+                final_vid = os.path.join(results_dir, f"concat_fim_{used_seed}.mp4")
+                partes_mp4_fade = self._gerar_lista_com_transicoes(pasta=results_dir, video_paths=partes_mp4, crossfade_frames=8)
+                self._concat_mp4s_no_reencode(partes_mp4_fade, final_vid)
+            else 
+                final_vid = partes_mp4[0]
             
             
             self._log_gpu_memory("Fim da Geração")