blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-04-06 03:57:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update 2-get-hubert-wav32k.py

Browse Source
This commit is contained in:
RVC-Boss 2024-01-25 23:28:08 +08:00 committed by GitHub
parent 56fef8a59c
commit d796bd40b9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
GPT_SoVITS/prepare_datasets/2-get-hubert-wav32k.py
View File

@ -47,7 +47,12 @@ os.makedirs(wav32dir,exist_ok=True)
maxx=0.95 maxx=0.95
alpha=0.5 alpha=0.5
device="cuda:0" if torch.cuda.is_available() else "mps" if torch.cuda.is_available():
device = "cuda:0"
elif torch.backends.mps.is_available():
device = "mps"
else:
device = "cpu"
model=cnhubert.get_model() model=cnhubert.get_model()
# is_half=False # is_half=False
if(is_half==True): if(is_half==True):
@ -106,4 +111,4 @@ if(len(nan_fails)>0 and is_half==True):
try: try:
name2go(wav_name) name2go(wav_name)
except: except:
print(wav_name,traceback.format_exc()) print(wav_name,traceback.format_exc())