Update api/ltx/ltx_aduc_pipeline.py

api/ltx/ltx_aduc_pipeline.py

@@ -563,7 +563,7 @@ class LtxAducPipeline:
 
     # (O resto das funções de _finalize_generation, _save_and_log_video, etc., permanecem as mesmas)
     @log_function_io
-    def _finalize_generation1(self, final_latents_cpu: torch.Tensor, base_filename: str, seed: int) -> Tuple[str, str]:
+    def _finalize_generation(self, final_latents_cpu: torch.Tensor, base_filename: str, seed: int) -> Tuple[str, str]:
         final_latents_path = RESULTS_DIR / f"latents_{base_filename}_{seed}.pt"
         torch.save(final_latents_cpu, final_latents_path)
         logging.info(f"Final latents saved to: {final_latents_path}")
@@ -609,7 +609,7 @@ class LtxAducPipeline:
         return random.randint(0, 2**32 - 1)
 
 
-    def _finalize_generation(self, latents_paths: List[Path], base_filename: str, seed: int) -> Tuple[str, str, int]:
+    def _finalize_generation1(self, latents_paths: List[Path], base_filename: str, seed: int) -> Tuple[str, str, int]:
         """Loads latents, concatenates, decodes to video, and saves both."""
         logging.info("Finalizing generation: decoding latents to video.")
         all_tensors_cpu = [torch.load(p) for p in latents_paths]