From 8aead029a8d8507dfafe16ed8d54e0a67df3e8a5 Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Fri, 17 Mar 2023 13:33:13 +0530 Subject: [PATCH] temp fix for installations that installed torch 2.0 by mistake --- scripts/on_sd_start.sh | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/scripts/on_sd_start.sh b/scripts/on_sd_start.sh index b9bc7950..b62a2fc2 100755 --- a/scripts/on_sd_start.sh +++ b/scripts/on_sd_start.sh @@ -65,8 +65,10 @@ 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 + # temp fix for installations that installed torch 2.0 by mistake + if [ "$OS_NAME" == "linux" ]; then + python -m pip install --upgrade torch==1.13.1+cu116 torchvision==0.14.1+cu116 --extra-index-url https://download.pytorch.org/whl/cu116 -q + elif [ "$OS_NAME" == "macos" ]; then python -m pip install --upgrade torch==1.13.1 torchvision==0.14.1 -q fi