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

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

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