Spaces:
Paused
Paused
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 ---
|