Update api/ltx/ltx_aduc_manager.py

api/ltx/ltx_aduc_manager.py

@@ -13,7 +13,8 @@ from pathlib import Path
 from typing import List, Optional, Tuple, Union, Dict
 from dataclasses import dataclass
 import threading
-
+import sys
+from pathlib import Path
 import torch
 from diffusers.utils.torch_utils import randn_tensor
 from huggingface_hub import hf_hub_download
@@ -22,12 +23,27 @@ from huggingface_hub import hf_hub_download
 from api.managers.gpu_manager import gpu_manager
 from api.ltx.ltx_utils import build_ltx_pipeline_on_cpu
 
+LTX_VIDEO_REPO_DIR = Path("/data/LTX-Video")
+LTX_REPO_ID = "Lightricks/LTX-Video"
+CACHE_DIR = os.environ.get("HF_HOME")
+
+
+def add_deps_to_path():
+    """
+    Adiciona o diretório do repositório LTX ao sys.path para garantir que suas
+    bibliotecas possam ser importadas.
+    """
+    repo_path = str(LTX_VIDEO_REPO_DIR.resolve())
+    if repo_path not in sys.path:
+        sys.path.insert(0, repo_path)
+        logging.info(f"[ltx_utils] LTX-Video repository added to sys.path: {repo_path}")
+
+# Executa a função imediatamente para configurar o ambiente antes de qualquer importação.
+add_deps_to_path()
+
 # --- Importações da biblioteca LTX-Video ---
-try:
-    from ltx_video.pipelines.pipeline_ltx_video import LTXVideoPipeline
-    from ltx_video.models.autoencoders.vae_encode import vae_encode, latent_to_pixel_coords
-except ImportError as e:
-    raise ImportError(f"Could not import from LTX-Video library. Check sys.path and repo integrity. Error: {e}")
+from ltx_video.pipelines.pipeline_ltx_video import LTXVideoPipeline
+from ltx_video.models.autoencoders.vae_encode import vae_encode, latent_to_pixel_coords
 
 
 # ==============================================================================