From 40c2fae6dcf1310323cf849dbaa5a8b7be677e79 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Stardust=C2=B7=E5=87=8F?= <star_dust_chen@foxmail.com>
Date: Sat, 9 Sep 2023 11:35:38 +0800
Subject: [PATCH] fix device

---
 webui.py | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/webui.py b/webui.py
index 348636c..b7b343a 100644
--- a/webui.py
+++ b/webui.py
@@ -1,8 +1,4 @@
 import sys, os
-
-if sys.platform == "darwin":
-    os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
-
 import logging
 
 logging.getLogger("numba").setLevel(logging.WARNING)
@@ -29,6 +25,14 @@ import webbrowser
 
 net_g = None
 
+if (
+    sys.platform == "darwin"
+    and torch.backends.mps.is_available()
+    ):
+    device = "mps"
+    os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
+else:
+    device = "cuda"
 
 def get_text(text, language_str, hps):
     norm_text, phone, tone, word2ph = clean_text(text, language_str)