Fix for blackwell GPUs on new installations

This commit is contained in:
cmdr2
2025-03-29 14:34:17 +05:30
parent 1b3d048bf2
commit 45c6865b2d
2 changed files with 4 additions and 2 deletions

View File

@ -46,7 +46,8 @@ fi
if [ -e "src" ]; then mv src src-old; fi
if [ -e "ldm" ]; then mv ldm ldm-old; fi
python -m pip install -q torchruntime
# this is outside check_modules.py to ensure that the required version of torchruntime is present
python -m pip install -q torchruntime>=1.15.1
cd ..
# Download the required packages