Update api/ltx/ltx_utils.py

api/ltx/ltx_utils.py

@@ -27,6 +27,8 @@ from huggingface_hub import logging as ll
 ll.set_verbosity_error()
 ll.set_verbosity_warning()
 ll.set_verbosity_info()
+from utils.debug_utils import log_function_io
+
 ll.set_verbosity_debug()
 
 
@@ -58,6 +60,7 @@ if repo_path not in sys.path:
 # --- FUNÇÕES DE CONSTRUÇÃO DE MODELO E PIPELINE ---
 # ==============================================================================
 
+@log_function_io
 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}")
@@ -66,6 +69,7 @@ def create_latent_upsampler(latent_upsampler_model_path: str, device: str) -> La
     latent_upsampler.eval()
     return latent_upsampler
 
+@log_function_io
 def build_ltx_pipeline_on_cpu(config: Dict) -> Tuple[LTXVideoPipeline, Optional[torch.nn.Module]]:
     """Builds the complete LTX pipeline and upsampler on the CPU."""
     t0 = time.perf_counter()
@@ -122,6 +126,7 @@ def build_ltx_pipeline_on_cpu(config: Dict) -> Tuple[LTXVideoPipeline, Optional[
 # --- FUNÇÕES AUXILIARES (Seed, Preparação de Imagem) ---
 # ==============================================================================
 
+@log_function_io
 def seed_everything(seed: int):
     """Sets the seed for reproducibility."""
     random.seed(seed)
@@ -132,6 +137,7 @@ def seed_everything(seed: int):
     torch.backends.cudnn.deterministic = True
     torch.backends.cudnn.benchmark = False
 
+@log_function_io
 def load_image_to_tensor_with_resize_and_crop(
     image_input: Union[str, Image.Image],
     target_height: int,