Update api/ltx_server.py

api/ltx_server.py

@@ -396,6 +396,38 @@ class VideoService:
         print(f"[DEBUG] Cond shape={tuple(out.shape)} dtype={out.dtype} device={out.device}")
         return out
 
+    def dividir_latentes(latents_brutos):
+    total = latents_brutos.shape[2]  # dimensão temporal (latentes)
+    
+    if total % 2 == 1:  # ÍMPAR
+        cut = total // 2
+        primeira = latents_brutos[:, :, :cut+1, :, :].clone()
+        segunda  = latents_brutos[:, :, cut:, :, :].clone()
+    else:  # PAR
+        cut = total // 2
+        # primeira parte até o meio, mas o último frame deve ser ajustado
+        primeira = latents_brutos[:, :, :cut+1, :, :].clone()
+        segunda  = latents_brutos[:, :, cut:, :, :].clone()
+
+    return primeira, segunda
+
+    def _concat_mp4s_no_reencode(self, mp4_a: str, mp4_b: str, out_path: str):
+    # Concat demuxer do ffmpeg (sem reencode)
+    import tempfile, subprocess, shlex, os
+    with tempfile.NamedTemporaryFile("w", delete=False, suffix=".txt") as f:
+        f.write(f"file '{os.path.abspath(mp4_a)}'\n")
+        f.write(f"file '{os.path.abspath(mp4_b)}'\n")
+        list_path = f.name
+    cmd = f"ffmpeg -y -f concat -safe 0 -i {list_path} -c copy {out_path}"
+    print(f"[DEBUG] Concat: {cmd}")
+    try:
+        subprocess.check_call(shlex.split(cmd))
+    finally:
+        try: os.remove(list_path)
+        except Exception: pass
+            
+    
+
     # --- 6. GERAÇÃO ---
     def generate(
         self,
@@ -569,40 +601,67 @@ class VideoService:
                 print(f"[DEBUG] Latentes (single-pass): shape={tuple(latents.shape)}")
 
             # Staging e escrita MP4 (simples: VAE → pixels → MP4)
+            
+            latents_cpu = latents.detach().to("cpu", non_blocking=True)
+            torch.cuda.empty_cache()
+            try:
+                torch.cuda.ipc_collect()
+            except Exception:
+                pass
+                
+            # 2) Divide em duas partes
+            lat_a, lat_b = dividir_latentes(latents_cpu)
+            print(f"[DEBUG] Partição A: {tuple(lat_a.shape)}")
+            print(f"[DEBUG] Partição B: {tuple(lat_b.shape)}")
+            
+            latents_parts = [lat_a, lat_b]
+            
             temp_dir = tempfile.mkdtemp(prefix="ltxv_"); self._register_tmp_dir(temp_dir)
             results_dir = "/app/output"; os.makedirs(results_dir, exist_ok=True)
-            output_video_path = os.path.join(temp_dir, f"output_{used_seed}.mp4")
-            final_output_path = None
-
-            print("[DEBUG] Decodificando bloco de latentes com VAE → tensor de pixels...")
-            # Usar manager com timestep por item; previne target_shape e rota NoneType.decode
-            pixel_tensor = vae_manager_singleton.decode(
-                latents.to(self.device, non_blocking=True),
-                decode_timestep=float(self.config.get("decode_timestep", 0.05))
-            )
-            log_tensor_info(pixel_tensor, "Pixel tensor (VAE saída)")
-
-            print("[DEBUG] Codificando MP4 a partir do tensor de pixels (bloco inteiro)...")
-            video_encode_tool_singleton.save_video_from_tensor(
-                pixel_tensor,
-                output_video_path,
-                fps=call_kwargs["frame_rate"],
-                progress_callback=progress_callback
-            )
-
-            candidate_final = os.path.join(results_dir, f"output_{used_seed}.mp4")
-            try:
-                shutil.move(output_video_path, candidate_final)
-                final_output_path = candidate_final
-                print(f"[DEBUG] MP4 movido para {final_output_path}")
-            except Exception as e:
-                final_output_path = output_video_path
-                print(f"[DEBUG] Falha no move; usando tmp como final: {e}")
 
-            self._register_tmp_file(output_video_path)
+            partes_mp4 = []
+            par = 0
+            
+            for latents in latents_parts:
+                par = par + 1
+                output_video_path = os.path.join(temp_dir, f"output_{used_seed}_{par}.mp4")
+                final_output_path = None
+    
+                print("[DEBUG] Decodificando bloco de latentes com VAE → tensor de pixels...")
+                # Usar manager com timestep por item; previne target_shape e rota NoneType.decode
+                pixel_tensor = vae_manager_singleton.decode(
+                    latents.to(self.device, non_blocking=True),
+                    decode_timestep=float(self.config.get("decode_timestep", 0.05))
+                )
+                log_tensor_info(pixel_tensor, "Pixel tensor (VAE saída)")
+    
+                print("[DEBUG] Codificando MP4 a partir do tensor de pixels (bloco inteiro)...")
+                video_encode_tool_singleton.save_video_from_tensor(
+                    pixel_tensor,
+                    output_video_path,
+                    fps=call_kwargs["frame_rate"],
+                    progress_callback=progress_callback
+                )
+    
+                candidate_final = os.path.join(results_dir, f"output_{used_seed}.mp4")
+                try:
+                    shutil.move(output_video_path, candidate_final)
+                    final_output_path = candidate_final
+                    print(f"[DEBUG] MP4 parte {par} movido para {final_output_path}")
+                    partes_mp4.append(final_output_path)
+                    
+                except Exception as e:
+                    final_output_path = output_video_path
+                    print(f"[DEBUG] Falha no move; usando tmp como final: {e}")
+
+            final_concat_tmp = os.path.join(temp_dir, f"concat_{used_seed}.mp4")
+            self._concat_mp4s_no_reencode(partes_mp4[0], partes_mp4[1], final_concat_tmp)
+
+            self._register_tmp_file(final_concat_tmp)
             self._log_gpu_memory("Fim da Geração")
             print(f"[DEBUG] generate() fim ok. total_time={time.perf_counter()-t_all:.3f}s")
-            return final_output_path, used_seed
+            return final_concat_tmp, used_seed
+            
 
         except Exception as e:
             print("[DEBUG] EXCEÇÃO NA GERAÇÃO:")