Spaces:
Paused
Paused
Update setup.py
Browse files
setup.py
CHANGED
|
@@ -74,8 +74,10 @@ def download_ltx_models():
|
|
| 74 |
"""
|
| 75 |
print("\n--- ETAPA 3: Verificando e baixando modelos LTX ---")
|
| 76 |
LTX_REPO_ID = "Lightricks/LTX-Video" # Repositório dos modelos no Hub
|
| 77 |
-
ltx_configs_dir = DEPS_DIR / "LTX-Video" / "configs"
|
| 78 |
-
|
|
|
|
|
|
|
| 79 |
if not ltx_configs_dir.is_dir():
|
| 80 |
print(f"ERRO CRÍTICO: Diretório de configs '{ltx_configs_dir}' não encontrado. O clone do LTX-Video pode ter falhado.")
|
| 81 |
sys.exit(1)
|
|
@@ -110,8 +112,8 @@ def download_ltx_models():
|
|
| 110 |
repo_id=LTX_REPO_ID,
|
| 111 |
filename=model_filename,
|
| 112 |
local_dir=str(MODELS_ROOT),
|
| 113 |
-
|
| 114 |
-
token=os.getenv("HF_TOKEN")
|
| 115 |
)
|
| 116 |
print(f"'{model_filename}' baixado com sucesso.")
|
| 117 |
except Exception as e:
|
|
@@ -123,7 +125,7 @@ def main():
|
|
| 123 |
|
| 124 |
# Executa as etapas na ordem correta
|
| 125 |
clone_repos()
|
| 126 |
-
install_ltx_dependencies()
|
| 127 |
download_ltx_models()
|
| 128 |
|
| 129 |
print("\n--- ✅ Setup do Ambiente Concluído com Sucesso! ---")
|
|
|
|
| 74 |
"""
|
| 75 |
print("\n--- ETAPA 3: Verificando e baixando modelos LTX ---")
|
| 76 |
LTX_REPO_ID = "Lightricks/LTX-Video" # Repositório dos modelos no Hub
|
| 77 |
+
#ltx_configs_dir = DEPS_DIR / "LTX-Video" / "configs"
|
| 78 |
+
ltx_configs_dir = "/app/configs"
|
| 79 |
+
|
| 80 |
+
|
| 81 |
if not ltx_configs_dir.is_dir():
|
| 82 |
print(f"ERRO CRÍTICO: Diretório de configs '{ltx_configs_dir}' não encontrado. O clone do LTX-Video pode ter falhado.")
|
| 83 |
sys.exit(1)
|
|
|
|
| 112 |
repo_id=LTX_REPO_ID,
|
| 113 |
filename=model_filename,
|
| 114 |
local_dir=str(MODELS_ROOT),
|
| 115 |
+
cache_dir=os.getenv("HF_HOME_CACHE"),
|
| 116 |
+
token=os.getenv("HF_TOKEN"),
|
| 117 |
)
|
| 118 |
print(f"'{model_filename}' baixado com sucesso.")
|
| 119 |
except Exception as e:
|
|
|
|
| 125 |
|
| 126 |
# Executa as etapas na ordem correta
|
| 127 |
clone_repos()
|
| 128 |
+
#install_ltx_dependencies()
|
| 129 |
download_ltx_models()
|
| 130 |
|
| 131 |
print("\n--- ✅ Setup do Ambiente Concluído com Sucesso! ---")
|