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

Update torch for ARM Macs to 2.3.1
This commit is contained in:
AUTOMATIC1111 2024-07-06 10:04:26 +03:00 committed by GitHub
commit 8058eed6a5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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
#################################################################### ####################################################################