Mirrors
/
stable-diffusion-webui
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update torch for ARM Macs to 2.3.1

This commit is contained in:
viking1304 2024-06-20 23:57:59 +02:00
parent a30b19dd55
commit a772fd9804
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
webui-macos-env.sh
View File

@ -16,7 +16,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
#################################################################### ####################################################################