Update on_sd_start.sh

Fix for weight load error
This commit is contained in:
Emrul 2025-03-11 03:53:41 +00:00 committed by GitHub
parent da163393ff
commit 2471d9ba21
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -52,6 +52,10 @@ cd ..
# Skip the package download and prompt if INSTALL_ONLY=1 is set
if [ "$INSTALL_ONLY" != "1" ]; then
# Download the required packages
# see https://github.com/easydiffusion/easydiffusion/issues/1911
export TORCH_FORCE_NO_WEIGHTS_ONLY_LOAD=1
python scripts/check_modules.py --launch-uvicorn
read -p "Press any key to continue"
else