Update inference_webui.py

Change inference device to accelerate inference on Mac and reduce memory leak
1 year ago · 861658050b
parent 3180294710
commit 861658050b
1 changed files with 1 additions and 1 deletions
--- a/GPT_SoVITS/inference_webui.py
+++ b/GPT_SoVITS/inference_webui.py
@ -73,7 +73,7 @@ os.environ['PYTORCH_ENABLE_MPS_FALLBACK'] = '1'  # 确保直接启动推理UI时
 if torch.cuda.is_available():
    device = "cuda"
 elif torch.backends.mps.is_available():
-    device = "mps"
+    device = "cpu"
 else:
    device = "cpu"