✨ add PYTORCH_ENABLE_MPS_FALLBACK

lenML · Jul 4, 2024 · 2024e0b · 2024e0b
1 parent a449091
commit 2024e0b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/webui.py b/webui.py
@@ -30,6 +30,9 @@
 dcls_patch()
 ignore_useless_warnings()
 
+if sys.platform == "darwin":
+    os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
+
 
 def setup_webui_args(parser: argparse.ArgumentParser):
     parser.add_argument("--server_name", type=str, help="server name")