Update cloning/clonevoice.py

cloning/clonevoice.py

@@ -3,7 +3,15 @@ from encodec.utils import convert_audio
 from bark.hubert.hubert_manager import HuBERTManager
 from bark.hubert.pre_kmeans_hubert import CustomHubert
 from bark.hubert.customtokenizer import CustomTokenizer
+import torch
+
+# Charge le modèle sur le CPU
+model = torch.load('chemin_vers_ton_modele.pth', map_location=torch.device('cpu'))
+model = model.eval()
 
+# Si vous avez besoin de déplacer le modèle vers un dispositif spécifique après le chargement
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+model = model.to(device)
 import torchaudio
 import torch
 import os