eeuuia commited on
Commit
8504d9f
·
verified ·
1 Parent(s): c75608c

Rename api/ltx_pool_manager.py to api/ltx/ltx_pool_manager.py

Browse files
api/{ltx_pool_manager.py → ltx/ltx_pool_manager.py} RENAMED
@@ -11,6 +11,19 @@ from diffusers.utils.torch_utils import randn_tensor
11
  # --- Importações da nossa arquitetura ---
12
  from api.gpu_manager import gpu_manager
13
  from api.ltx.ltx_utils import build_ltx_pipeline_on_cpu
 
 
 
 
 
 
 
 
 
 
 
 
 
14
  from ltx_video.pipelines.pipeline_ltx_video import LTXVideoPipeline
15
 
16
  # --- Definição dos nossos Data Classes ---
 
11
  # --- Importações da nossa arquitetura ---
12
  from api.gpu_manager import gpu_manager
13
  from api.ltx.ltx_utils import build_ltx_pipeline_on_cpu
14
+
15
+ def add_deps_to_path():
16
+ """
17
+ Adiciona o diretório do repositório LTX ao sys.path para garantir que suas
18
+ bibliotecas possam ser importadas.
19
+ """
20
+ repo_path = str(LTX_VIDEO_REPO_DIR.resolve())
21
+ if repo_path not in sys.path:
22
+ sys.path.insert(0, repo_path)
23
+ logging.info(f"[ltx_utils] LTX-Video repository added to sys.path: {repo_path}")
24
+
25
+ # Executa a função imediatamente para configurar o ambiente antes de qualquer importação.
26
+ add_deps_to_path()
27
  from ltx_video.pipelines.pipeline_ltx_video import LTXVideoPipeline
28
 
29
  # --- Definição dos nossos Data Classes ---