From bce9eb02512e02757d0b2f9d99369087c0752c74 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=BA=90=E6=96=87=E9=9B=A8?=
 <41315874+fumiama@users.noreply.github.com>
Date: Sun, 3 Sep 2023 17:04:42 +0800
Subject: [PATCH] fix: mps

---
 webui.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/webui.py b/webui.py
index 58cea49..9a7d5af 100644
--- a/webui.py
+++ b/webui.py
@@ -1,3 +1,8 @@
+import sys, os
+
+if sys.platform == "darwin":
+    os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
+
 import torch
 import argparse
 import commons
@@ -8,10 +13,7 @@ from text import cleaned_text_to_sequence, get_bert
 from text.cleaner import clean_text
 import gradio as gr
 import webbrowser
-import sys, os
 
-if sys.platform == "darwin":
-    os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
 
 def get_text(text, language_str, hps):
     norm_text, phone, tone, word2ph = clean_text(text, language_str)