Update 2-get-hubert-wav32k.py

2 years ago · d796bd40b9
parent 56fef8a59c
commit d796bd40b9
1 changed files with 7 additions and 2 deletions
--- a/GPT_SoVITS/prepare_datasets/2-get-hubert-wav32k.py
+++ b/GPT_SoVITS/prepare_datasets/2-get-hubert-wav32k.py
@ -47,7 +47,12 @@ os.makedirs(wav32dir,exist_ok=True)

 maxx=0.95
 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()
 # is_half=False
 if(is_half==True):