Update deformes4D_engine.py

deformes4D_engine.py

@@ -264,32 +264,31 @@ class Deformes4DEngine:
 
 
     def _render_and_post_process(self, final_concatenated_latents: torch.Tensor, 
-                                 base_name: str, expected_height: int, expected_width: int, fps: int = 24) -> str:
-        """
-        Aplica upscale + refinamento nos latentes finais e salva o vídeo em alta qualidade.
-        """
+                             base_name: str, expected_height: int, expected_width: int, fps: int = 24) -> str:
         logger.info("Iniciando pós-processamento: upscale + refinamento...")
 
-        # --- 1. Upscale espacial com LTX Upscaler ---
+        # --- 1. Upscale ---
         upscaled_latents = upscaler_specialist_singleton.upscale(final_concatenated_latents)
-
         logger.info(f"Upscale concluído: shape {list(upscaled_latents.shape)}")
 
-        # --- 2. Refinamento difusivo curto ---
+        # --- 2. Refinamento ---
+        _, _, _, h, w = upscaled_latents.shape
         refined_latents, _ = ltx_manager_singleton.refine_latents(
             upscaled_latents,
+            height=h,
+            width=w,
             denoise_strength=0.4,
             refine_steps=10
         )
         logger.info("Refinamento concluído.")
 
-        # --- 3. Decodificação para pixels ---
+        # --- 3. Decodificação ---
         pixel_tensor = self.latents_to_pixels(refined_latents)
         video_path = os.path.join(self.workspace_dir, f"{base_name}_HQ.mp4")
         self.save_video_from_tensor(pixel_tensor, video_path, fps=fps)
-
         logger.info(f"Vídeo final salvo em: {video_path}")
-        return video_path
+    return refined_latents
+
         
     
     def _generate_latent_tensor_internal(self, conditioning_items, ltx_params, target_resolution, total_frames_to_generate):