tuna2134/sbv2-v2
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

fix device

Browse Source
This commit is contained in:
Stardust·减
2023-09-09 11:35:38 +08:00
committed by GitHub
parent a6c42d6fbc
commit 40c2fae6dc
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
webui.py
View File

@@ -1,8 +1,4 @@
import sys, os import sys, os
if sys.platform == "darwin":
os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
import logging import logging
logging.getLogger("numba").setLevel(logging.WARNING) logging.getLogger("numba").setLevel(logging.WARNING)
@@ -29,6 +25,14 @@ import webbrowser
net_g = None 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): def get_text(text, language_str, hps):
norm_text, phone, tone, word2ph = clean_text(text, language_str) norm_text, phone, tone, word2ph = clean_text(text, language_str)