Update api/ltx_server.py

api/ltx_server.py

@@ -396,6 +396,41 @@ class VideoService:
         print(f"[DEBUG] Cond shape={tuple(out.shape)} dtype={out.dtype} device={out.device}")
         return out
 
+
+    def dividir_latentes_em_partes(latents_brutos, quantidade: int):
+        """
+        Divide um tensor de latentes em `quantidade` partes e retorna uma lista de clones.
+        
+        Args:
+            latents_brutos: tensor [B, C, T, H, W]
+            quantidade: número de partes que queremos dividir
+        
+        Returns:
+            List[Tensor]: lista de `quantidade` partes, cada uma cloneada
+        """
+        total = latents_brutos.shape[2]  # dimensão temporal
+        partes = []
+    
+        if quantidade <= 1 or quantidade > total:
+            return [latents_brutos.clone()]
+    
+        # calcular tamanho aproximado de cada parte
+        step = total // quantidade
+        overlap = 1  # sobreposição mínima de 1 frame entre partes
+    
+        for i in range(quantidade):
+            start = i * step
+            end = start + step
+            if i == quantidade - 1:
+                end = total  # última parte vai até o final
+            else:
+                end += overlap  # sobreposição
+            parte = latents_brutos[:, :, start:end, :, :].clone()
+            partes.append(parte)
+    
+        return partes
+    
+
     def _dividir_latentes(self, latents_brutos):
         total = latents_brutos.shape[2]  # dimensão temporal (número de latentes)
         
@@ -609,7 +644,7 @@ class VideoService:
                 pass
                 
             # 2) Divide em duas partes
-            lat_a, lat_b = self._dividir_latentes(latents_cpu)
+            lat_a, lat_b = self._dividir_latentes_em_partes(latents_cpu, 2)
             print(f"[DEBUG] Partição A: {tuple(lat_a.shape)}")
             print(f"[DEBUG] Partição B: {tuple(lat_b.shape)}")
             
@@ -653,18 +688,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_{used_seed}.mp4")
+            final_concat = os.path.join(results_dir, f"concat_fim_{used_seed}.mp4")
             self._concat_mp4s_no_reencode(partes_mp4[0], partes_mp4[1], final_concat)
 
-            try:
-                shutil.move(output_video_path, final_concat)
-                final_output_path = final_concat
-            except Exception:
-                final_output_path = output_video_path
-            self._register_tmp_file(output_video_path)
-
+            
             self._log_gpu_memory("Fim da Geração")
-            return final_output_path, used_seed
+            return final_concat, used_seed
             
 
         except Exception as e: