From a772fd9804944cc19c4d6a03ccfbaa6066ce62a8 Mon Sep 17 00:00:00 2001
From: viking1304 <viking1304@gmail.com>
Date: Thu, 20 Jun 2024 23:57:59 +0200
Subject: [PATCH] Update torch for ARM Macs to 2.3.1

---
 webui-macos-env.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/webui-macos-env.sh b/webui-macos-env.sh
index ad0736378..f390e4d34 100644
--- a/webui-macos-env.sh
+++ b/webui-macos-env.sh
@@ -16,7 +16,7 @@ export PYTORCH_ENABLE_MPS_FALLBACK=1
 if [[ "$(sysctl -n machdep.cpu.brand_string)" =~ ^.*"Intel".*$ ]]; then
     export TORCH_COMMAND="pip install torch==2.1.2 torchvision==0.16.2"
 else
-    export TORCH_COMMAND="pip install torch==2.3.0 torchvision==0.18.0"
+    export TORCH_COMMAND="pip install torch==2.3.1 torchvision==0.18.1"
 fi
 
 ####################################################################