blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-04-05 19:41:56 +08:00
Code Issues Packages Projects Releases Wiki Activity

mps support, optimized device selection

Browse Source
This commit is contained in:
Wu Zichen 2024-01-24 16:45:13 +08:00
parent aed4935fce
commit 382102c9d0
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
config.py
View File

@ -1,5 +1,6 @@
import sys,os
import torch
# 推理用的指定模型
sovits_path = ""
@ -14,7 +15,12 @@ pretrained_gpt_path = "GPT_SoVITS/pretrained_models/s1bert25hz-2kh-longer-epoch=
exp_root = "logs"
python_exec = sys.executable or "python"
infer_device = "cuda"
if torch.cuda.is_available():
infer_device = "cuda"
elif torch.mps.is_available():
infer_device = "mps"
else:
infer_device = "cpu"
webui_port_main = 9874
webui_port_uvr5 = 9873