aducsdr commited on
Commit
91b2426
·
verified ·
1 Parent(s): 35a21c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -17,10 +17,10 @@ import os
17
  import sys
18
 
19
  # --- ETAPA 1: Clonar o Repositório do GitHub ---
20
- repo_name = "SeedVR2-3B"
21
  if not os.path.exists(repo_name):
22
  print(f"Clonando o repositório {repo_name} do GitHub...")
23
- subprocess.run(f"git clone https://huggingface.co/spaces/ByteDance-Seed/SeedVR2-3B", shell=True, check=True)
24
 
25
  # --- ETAPA 2: Mudar para o Diretório e Configurar o Ambiente ---
26
  os.chdir(repo_name)
@@ -50,7 +50,7 @@ subprocess.run(
50
  print("✅ Configuração do flash-attn concluída.")
51
 
52
 
53
- # Instalar Apex
54
  print("Instalando Apex a partir do código-fonte...")
55
  subprocess.run(
56
  [
@@ -59,7 +59,7 @@ subprocess.run(
59
  "pip",
60
  "install",
61
  "--no-build-isolation",
62
- "--no-pep517",
63
  "git+https://github.com/NVIDIA/apex.git"
64
  ],
65
  check=True
 
17
  import sys
18
 
19
  # --- ETAPA 1: Clonar o Repositório do GitHub ---
20
+ repo_name = "SeedVR"
21
  if not os.path.exists(repo_name):
22
  print(f"Clonando o repositório {repo_name} do GitHub...")
23
+ subprocess.run(f"git clone https://github.com/ByteDance-Seed/{repo_name}.git", shell=True, check=True)
24
 
25
  # --- ETAPA 2: Mudar para o Diretório e Configurar o Ambiente ---
26
  os.chdir(repo_name)
 
50
  print("✅ Configuração do flash-attn concluída.")
51
 
52
 
53
+ # Instalar Apex, forçando o uso do setup.py para compilar as extensões CUDA
54
  print("Instalando Apex a partir do código-fonte...")
55
  subprocess.run(
56
  [
 
59
  "pip",
60
  "install",
61
  "--no-build-isolation",
62
+ "--no-pep517", # Força o uso do setup.py para compilar as extensões CUDA
63
  "git+https://github.com/NVIDIA/apex.git"
64
  ],
65
  check=True