Update api/ltx_server.py

api/ltx_server.py

@@ -850,53 +850,60 @@ class VideoService:
             # --- ETAPA FINAL: DECODIFICAÇÃO E CODIFICAÇÃO MP4 ---
             print("\n--- INICIANDO ETAPA FINAL: DECODIFICAÇÃO E MONTAGEM ---")
             
-            #latents_cpu = latents.detach().to("cpu", non_blocking=True)
-            #torch.cuda.empty_cache()
-            #try:
-            #    torch.cuda.ipc_collect()
-            #except Exception:
-            #    pass
-
-            latents_parts = []
-            for latents in latents_list:
-                latents_parts.append(self._dividir_latentes_por_tamanho(latents,15,1))
-                
-           
+            temp_dir = tempfile.mkdtemp(prefix="ltxv_"); self._register_tmp_dir(temp_dir)
+            results_dir = "/app/output"; os.makedirs(results_dir, exist_ok=True)
+    
+            latents_parts_up = self._dividir_latentes_por_tamanho(latents_list,15,1)
+
             partes_mp4 = []
             par = 0
-            for latents in latents_parts:
-                
-                par = par + 1
-                output_video_path = os.path.join(results_dir, f"output_{used_seed}_{par}.mp4")
-                final_output_path = None
-    
-                print("[DEBUG] Decodificando bloco de latentes com VAE {par} → 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 = os.path.join(results_dir, f"output_par_{par}.mp4")
+            
+            for latents_vae in latents_parts_up:
+            
+                latents_cpu_vae = latents_vae.detach().to("cpu", non_blocking=True)
+                torch.cuda.empty_cache()
                 try:
-                    shutil.move(output_video_path, candidate)
-                    final_output_path = candidate
-                    print(f"[DEBUG] MP4 parte {par} movido para {final_output_path}")
-                    partes_mp4.append(final_output_path)
+                    torch.cuda.ipc_collect()
+                except Exception:
+                    pass
                     
-                except Exception as e:
-                    final_output_path = output_video_path
-                    print(f"[DEBUG] Falha no move; usando tmp como final: {e}")
-
+                latents_parts_vae = self._dividir_latentes_por_tamanho(latents_cpu_vae,15,1)
+                
+                for latents in latents_parts_vae:
+                    print(f"[DEBUG] Partição {par}: {tuple(latents.shape)}")
+           
+                    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 = os.path.join(results_dir, f"output_par_{par}.mp4")
+                    try:
+                        shutil.move(output_video_path, candidate)
+                        final_output_path = candidate
+                        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}")
+    
+            
             total_partes = len(partes_mp4)
             if (total_partes>1):
                 final_vid = os.path.join(results_dir, f"concat_fim_{used_seed}.mp4")