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 inference_webui.py

Browse Source 
Change inference device to accelerate inference on Mac and reduce memory leak
This commit is contained in:
XXXXRT666 2024-02-20 16:03:08 +00:00 committed by GitHub
parent 3180294710
commit 861658050b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
GPT_SoVITS/inference_webui.py
View File

@ -73,7 +73,7 @@ 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(): elif torch.backends.mps.is_available():
device = "mps" device = "cpu"
else: else:
device = "cpu" device = "cpu"