Merge pull request #16059 from viking1304/mac-torch-231

Update torch for ARM Macs to 2.3.1

Merge pull request #16059 from viking1304/mac-torch-231
Update torch for ARM Macs to 2.3.1
8058eed6 · AUTOMATIC1111 · GitHub · 3bd4a081 · a772fd98 · 8058eed6
Commit 8058eed6 authored Jul 06, 2024 by AUTOMATIC1111 Committed by GitHub Jul 06, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

webui-macos-env.sh webui-macos-env.sh +1 -1

No files found.
--- a/webui-macos-env.sh
+++ b/webui-macos-env.sh
@@ -11,7 +11,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

 ####################################################################