From 7a6d251e44cc4468c1ae9f8b8ddd1de8c1b2dafa Mon Sep 17 00:00:00 2001
From: chasonjiang <1440499136@qq.com>
Date: Mon, 11 Mar 2024 17:31:21 +0800
Subject: [PATCH] 	modified:   GPT_SoVITS/inference_webui.py

---
 GPT_SoVITS/inference_webui.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/GPT_SoVITS/inference_webui.py b/GPT_SoVITS/inference_webui.py
index 53333c4..5a26615 100644
--- a/GPT_SoVITS/inference_webui.py
+++ b/GPT_SoVITS/inference_webui.py
@@ -45,9 +45,11 @@ os.environ['PYTORCH_ENABLE_MPS_FALLBACK'] = '1'  # 确保直接启动推理UI时
 
 if torch.cuda.is_available():
     device = "cuda"
+elif torch.backends.mps.is_available():
+    device = "mps"
 else:
     device = "cpu"
-
+    
 dict_language = {
     i18n("中文"): "all_zh",#全部按中文识别
     i18n("英文"): "en",#全部按英文识别#######不变