extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2024-12-25 16:38:55 +01:00
Code Issues Packages Projects Releases Wiki Activity

temp fix for installations that installed torch 2.0 by mistake

Browse Source
This commit is contained in:
cmdr2 2023-03-17 13:33:13 +05:30
parent a9da65c2cd
commit 8aead029a8
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/on_sd_start.sh
View File

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