From 302e2f9f4430b61c3dd81ddff96d7c23cf2fac37 Mon Sep 17 00:00:00 2001
From: bugyalu <774933704@qq.com>
Date: Wed, 10 May 2023 00:06:30 +0800
Subject: [PATCH] fix(#824): can not start the webui on Mac Arm64

---
 installer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/installer.py b/installer.py
index 2da153806..15ca2074c 100644
--- a/installer.py
+++ b/installer.py
@@ -211,7 +211,7 @@ def check_torch():
             xformers_package = os.environ.get('XFORMERS_PACKAGE', 'none')
         else:
             log.info('Using CPU-only Torch')
-            torch_command = os.environ.get('TORCH_COMMAND', 'torch torchaudio torchvision')
+            torch_command = os.environ.get('TORCH_COMMAND', 'torch==2.0.0 torchaudio torchvision')
             xformers_package = os.environ.get('XFORMERS_PACKAGE', 'none')
     if 'torch' in torch_command and not args.version:
         install(torch_command, 'torch torchvision torchaudio')