From 45c6865b2dda9375c0fabd7e314d6d7965453ed3 Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Sat, 29 Mar 2025 14:34:17 +0530 Subject: [PATCH] Fix for blackwell GPUs on new installations --- scripts/on_sd_start.bat | 3 ++- scripts/on_sd_start.sh | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/scripts/on_sd_start.bat b/scripts/on_sd_start.bat index f9113d81..ba619b80 100644 --- a/scripts/on_sd_start.bat +++ b/scripts/on_sd_start.bat @@ -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 diff --git a/scripts/on_sd_start.sh b/scripts/on_sd_start.sh index e491ad2d..c8edf2ed 100755 --- a/scripts/on_sd_start.sh +++ b/scripts/on_sd_start.sh @@ -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