Fix for blackwell GPUs on new installations

This commit is contained in:
cmdr2 2025-03-29 14:34:17 +05:30
parent 4044d696e6
commit 5c7625c425
2 changed files with 4 additions and 2 deletions

View File

@ -71,7 +71,8 @@ echo PYTHONPATH=%PYTHONPATH%
call where python
call python --version
call python -m pip install -q torchruntime
@rem this is outside check_modules.py to ensure that the required version of torchruntime is present
call python -m pip install -q torchruntime>=1.15.1
call python scripts\check_modules.py --launch-uvicorn
pause

View File

@ -50,7 +50,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