Update api/ltx/ltx_aduc_pipeline.py

api/ltx/ltx_aduc_pipeline.py

@@ -54,7 +54,7 @@ add_deps_to_path()
 
 # --- PROJECT IMPORTS ---
 try:
-    from ltx_video.pipelines.pipeline_ltx_video import LTXVideoPipeline, create_latent_upsampler # E outros...
+    from ltx_video.pipelines.pipeline_ltx_video import LTXVideoPipeline # E outros...
     from ltx_video.models.autoencoders.causal_video_autoencoder import CausalVideoAutoencoder
     from ltx_video.models.transformers.transformer3d import Transformer3DModel
     from ltx_video.models.transformers.symmetric_patchifier import SymmetricPatchifier
@@ -63,7 +63,7 @@ try:
     from safetensors import safe_open
     from api.gpu_manager import gpu_manager
     from ltx_video.models.autoencoders.vae_encode import (normalize_latents, un_normalize_latents)
-    from ltx_video.pipelines.pipeline_ltx_video import (ConditioningItem, LTXMultiScalePipeline, adain_filter_latent, create_latent_upsampler)
+    from ltx_video.pipelines.pipeline_ltx_video import (ConditioningItem, LTXMultiScalePipeline, adain_filter_latent)
     from ltx_video.utils.inference_utils import load_image_to_tensor_with_resize_and_crop
     from managers.vae_manager import vae_manager_singleton
     from tools.video_encode_tool import video_encode_tool_singleton
@@ -248,7 +248,14 @@ class LtxAducPipeline:
 
         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."""