diff --git a/scripts/on_sd_start.bat b/scripts/on_sd_start.bat index 2d5ff0ba..bee3b0db 100644 --- a/scripts/on_sd_start.bat +++ b/scripts/on_sd_start.bat @@ -71,6 +71,8 @@ echo PYTHONPATH=%PYTHONPATH% call where python call python --version +call python -m pip install -q torchruntime + call python scripts\check_modules.py --launch-uvicorn pause exit /b diff --git a/scripts/on_sd_start.sh b/scripts/on_sd_start.sh index a9fc809d..0e9f1a08 100755 --- a/scripts/on_sd_start.sh +++ b/scripts/on_sd_start.sh @@ -50,6 +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 + cd .. # Download the required packages python scripts/check_modules.py --launch-uvicorn