diff --git a/scripts/on_sd_start.sh b/scripts/on_sd_start.sh index 4b1b7ab5..b9bc7950 100755 --- a/scripts/on_sd_start.sh +++ b/scripts/on_sd_start.sh @@ -65,6 +65,11 @@ esac # install torch and torchvision if python ../scripts/check_modules.py torch torchvision; then + # temp fix for macOS installations that installed torch 2.0 by mistake + if [ "$OS_NAME" == "macos" ]; then + python -m pip install --upgrade torch==1.13.1 torchvision==0.14.1 -q + fi + echo "torch and torchvision have already been installed." else echo "Installing torch and torchvision.."