Spaces:
Paused
Paused
Update app.py
Browse files
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 = "
|
| 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://
|
| 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
|