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

Merge pull request #557 from XXXXRT666/main

Browse Source 
Change the inference device on Mac
This commit is contained in:
RVC-Boss 2024-02-21 16:27:42 +08:00 committed by GitHub
commit 7ee72bec7e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
GPT_SoVITS/inference_webui.py
View File

@ -72,8 +72,6 @@ os.environ['PYTORCH_ENABLE_MPS_FALLBACK'] = '1' # 确保直接启动推理UI时
if torch.cuda.is_available(): if torch.cuda.is_available():
device = "cuda" device = "cuda"
elif torch.backends.mps.is_available():
device = "mps"
else: else:
device = "cpu" device = "cpu"

2
config.py
View File

@ -19,8 +19,6 @@ exp_root = "logs"
python_exec = sys.executable or "python" python_exec = sys.executable or "python"
if torch.cuda.is_available(): if torch.cuda.is_available():
infer_device = "cuda" infer_device = "cuda"
elif torch.backends.mps.is_available():
infer_device = "mps"
else: else:
infer_device = "cpu" infer_device = "cpu"