Commit 8058eed6 authored by AUTOMATIC1111's avatar AUTOMATIC1111 Committed by GitHub

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

Update torch for ARM Macs to 2.3.1
parents 3bd4a081 a772fd98
...@@ -11,7 +11,7 @@ export PYTORCH_ENABLE_MPS_FALLBACK=1 ...@@ -11,7 +11,7 @@ export PYTORCH_ENABLE_MPS_FALLBACK=1
if [[ "$(sysctl -n machdep.cpu.brand_string)" =~ ^.*"Intel".*$ ]]; then if [[ "$(sysctl -n machdep.cpu.brand_string)" =~ ^.*"Intel".*$ ]]; then
export TORCH_COMMAND="pip install torch==2.1.2 torchvision==0.16.2" export TORCH_COMMAND="pip install torch==2.1.2 torchvision==0.16.2"
else 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 fi
#################################################################### ####################################################################
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment