File size: 1,936 Bytes
f2f8f13 87cd072 08c1163 87cd072 92e3233 87cd072 08c1163 87cd072 08c1163 87cd072 08c1163 87cd072 08c1163 87cd072 08c1163 87cd072 08c1163 87cd072 08c1163 e8e9070 08c1163 87cd072 08c1163 87cd072 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 |
import os
import subprocess
import sys
# Cloner votre repository
if not os.path.exists("VV_Clone_1"):
subprocess.run(["git", "clone", "https://github.com/Socold/VV_Clone_1.git"])
sys.path.insert(0, "VV_Clone_1")
# Installer les dépendances
subprocess.run([sys.executable, "-m", "pip", "install", "-q",
"torch", "transformers>=4.40.0", "accelerate", "gradio>=4.0.0",
"soundfile", "librosa", "scipy", "numpy", "ffmpeg-python"])
subprocess.run([sys.executable, "-m", "pip", "install", "-e", "VV_Clone_1"])
import torch
import gradio as gr
# Configuration - Changé pour utiliser 7B
MODEL_PATH = "aoi-ot/VibeVoice-7B" # ← Modifié ici
# Récupérer les secrets (invisibles publiquement)
USERNAME = os.environ.get("AUTH_USERNAME")
PASSWORD = os.environ.get("AUTH_PASSWORD")
print(f"Chargement du modèle: {MODEL_PATH}")
print(f"GPU: {torch.cuda.get_device_name(0) if torch.cuda.is_available() else 'CPU'}")
print(f"VRAM disponible: {torch.cuda.get_device_properties(0).total_memory / 1024**3:.2f} GB")
# Lancer le script principal avec les arguments
os.chdir("VV_Clone_1")
sys.argv = ["gradio_demo.py", "--model_path", MODEL_PATH, "--share"]
# Importer et modifier le main pour ajouter l'auth
from demo.gradio_demo import main
# Si les secrets existent, modifier le fichier pour ajouter l'auth
if USERNAME and PASSWORD:
with open("demo/gradio_demo.py", "r") as f:
content = f.read()
# Remplacer la ligne launch pour ajouter auth
content = content.replace(
'demo.launch(share=args.share)',
f'demo.launch(share=args.share, auth=("{USERNAME}", "{PASSWORD}"))'
)
with open("demo/gradio_demo.py", "w") as f:
f.write(content)
print(f"Authentification activée pour l'utilisateur: {USERNAME}")
else:
print("Pas d'authentification configurée (pas de secrets définis)")
# Lancer le main
print("Démarrage de l'interface VibeVoice-7B...")
main() |