diff --git a/scripts/on_sd_start.bat b/scripts/on_sd_start.bat index 57c3341b..1fe15934 100644 --- a/scripts/on_sd_start.bat +++ b/scripts/on_sd_start.bat @@ -149,8 +149,12 @@ call WHERE uvicorn > .tmp for %%J in ("sd-v1-4.ckpt") do if "%%~zJ" EQU "7703807346" ( echo "Data files (weights) necessary for Stable Diffusion were already downloaded" ) else ( - echo. & echo "The model file present at %cd%\sd-v1-4.ckpt is invalid. It is only %%~zJ bytes in size. Re-downloading.." & echo. - del "sd-v1-4.ckpt" + for %%K in ("sd-v1-4.ckpt") do if "%%~zK" EQU "7703810927" ( + echo "Data files (weights) necessary for Stable Diffusion were already downloaded" + ) else ( + echo. & echo "The model file present at %cd%\sd-v1-4.ckpt is invalid. It is only %%~zK bytes in size. Re-downloading.." & echo. + del "sd-v1-4.ckpt" + ) ) ) ) diff --git a/scripts/on_sd_start.sh b/scripts/on_sd_start.sh index a9fd9240..87621258 100755 --- a/scripts/on_sd_start.sh +++ b/scripts/on_sd_start.sh @@ -162,7 +162,7 @@ if [ ! -f "sd-v1-4.ckpt" ]; then if [ -f "sd-v1-4.ckpt" ]; then model_size=`ls -l sd-v1-4.ckpt | awk '{print $5}'` - if [ ! "$model_size" -eq "4265380512" ] && [ ! "$model_size" -eq "7703807346" ]; then + if [ ! "$model_size" -eq "4265380512" ] && [ ! "$model_size" -eq "7703807346" ] && [ ! "$model_size" -eq "7703810927" ]; then printf "\n\nError: The downloaded model file was invalid! Bytes downloaded: $model_size\n\n" printf "\n\nError downloading the data files (weights) for Stable Diffusion. Sorry about that, please try to:\n 1. Run this installer again.\n 2. If that doesn't fix it, please try the common troubleshooting steps at https://github.com/cmdr2/stable-diffusion-ui/blob/main/Troubleshooting.md\n 3. If those steps don't help, please copy *all* the error messages in this window, and ask the community at https://discord.com/invite/u9yhsFmEkB\n 4. If that doesn't solve the problem, please file an issue at https://github.com/cmdr2/stable-diffusion-ui/issues\nThanks!\n\n" read -p "Press any key to continue"