diff --git a/scripts/on_sd_start.bat b/scripts/on_sd_start.bat index dc9b2417..4974f1b3 100644 --- a/scripts/on_sd_start.bat +++ b/scripts/on_sd_start.bat @@ -78,7 +78,7 @@ call WHERE uvicorn > .tmp ) @if exist "sd-v1-4.ckpt" ( - for %%I in ("sd-v1-4.ckpt") do if %%~zI GTR 400000000 ( + for %%I in ("sd-v1-4.ckpt") do if "%%~zI" EQU "4265380512" ( 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 %%~zI bytes in size. Re-downloading.." & echo. @@ -92,7 +92,7 @@ call WHERE uvicorn > .tmp @call curl -L -k https://me.cmdr2.org/stable-diffusion-ui/sd-v1-4.ckpt > sd-v1-4.ckpt @if exist "sd-v1-4.ckpt" ( - for %%I in ("sd-v1-4.ckpt") do if %%~zI LSS 400000000 ( + for %%I in ("sd-v1-4.ckpt") do if "%%~zI" NEQ "4265380512" ( echo. & echo "Error: The downloaded model file was invalid! Bytes downloaded: %%~zI" & echo. echo. & echo "Error downloading the data files (weights) for Stable Diffusion. Please try re-running this installer. If it doesn't work, please copy the messages in this window, and ask the community at https://discord.com/invite/u9yhsFmEkB or file an issue at https://github.com/cmdr2/stable-diffusion-ui/issues" & echo. pause diff --git a/scripts/on_sd_start.sh b/scripts/on_sd_start.sh index 538afce9..c3ee44ba 100755 --- a/scripts/on_sd_start.sh +++ b/scripts/on_sd_start.sh @@ -76,7 +76,7 @@ fi if [ -f "sd-v1-4.ckpt" ]; then model_size=`ls -l sd-v1-4.ckpt | awk '{print $5}'` - if [ "$model_size" -gt "4000000000" ]; then + if [ "$model_size" -eq "4265380512" ]; then echo "Data files (weights) necessary for Stable Diffusion were already downloaded" else printf "\n\nThe model file present at $PWD/sd-v1-4.ckpt is invalid. It is only $model_size bytes in size. Re-downloading.." @@ -91,7 +91,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" -lt "4000000000" ]; then + if [ ! "$model_size" -eq "4265380512" ]; 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. Please try re-running this installer. If it doesn't work, please copy the messages in this window, and ask the community at https://discord.com/invite/u9yhsFmEkB or file an issue at https://github.com/cmdr2/stable-diffusion-ui/issues\n\n" read -p "Press any key to continue"