Update api/ltx/ltx_aduc_pipeline.py

api/ltx/ltx_aduc_pipeline.py

@@ -164,6 +164,14 @@ class LtxAducPipeline:
     # --- LIFECYCLE & MODEL MANAGEMENT ---
     # ==========================================================================
 
+    def _create_latent_upsampler(self, latent_upsampler_model_path: str, device: str) -> LatentUpsampler:
+        """Loads the Latent Upsampler model from a checkpoint path."""
+        logging.info(f"Loading Latent Upsampler from: {latent_upsampler_model_path} to device: {device}")
+        latent_upsampler = LatentUpsampler.from_pretrained(latent_upsampler_model_path)
+        latent_upsampler.to(device)
+        latent_upsampler.eval()
+        return latent_upsampler
+    
     def _load_config(self) -> Dict:
         """Loads the YAML configuration file."""
         config_path = DEFAULT_CONFIG_FILE
@@ -242,20 +250,13 @@ class LtxAducPipeline:
             spatial_path = Path(spatial_path_str)
             if not ckpt_path.is_file():
                 raise FileNotFoundError(f"Main checkpoint file not found: {spatial_path}")
-            latent_upsampler = create_latent_upsampler(spatial_path, device="cpu")
+            latent_upsampler = self._create_latent_upsampler(spatial_path, device="cpu")
             if precision == "bfloat16":
                 latent_upsampler.to(torch.bfloat16)
 
         logging.info(f"Modelos LTX carregados na CPU em {time.perf_counter()-t0:.2f}s")
         return pipeline, latent_upsampler
 
-    def create_latent_upsampler(latent_upsampler_model_path: str, device: str) -> LatentUpsampler:
-        """Loads the Latent Upsampler model from a checkpoint path."""
-        logging.info(f"Loading Latent Upsampler from: {latent_upsampler_model_path} to device: {device}")
-        latent_upsampler = LatentUpsampler.from_pretrained(latent_upsampler_model_path)
-        latent_upsampler.to(device)
-        latent_upsampler.eval()
-        return latent_upsampler
     
     def move_to_device(self, main_device_str: str, vae_device_str: str):
         """Moves pipeline components to their target devices."""