Eueuiaa commited on
Commit
a07fd79
·
verified ·
1 Parent(s): cb400e1

Update setup.py

Browse files
Files changed (1) hide show
  1. setup.py +7 -5
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
- local_dir_use_symlinks=False,
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! ---")