Allow using waifu 7 GB weights

This commit is contained in:
cmdr2 2022-09-12 20:44:54 +05:30
parent a32a58bd0f
commit a955730086
2 changed files with 7 additions and 3 deletions

View File

@ -149,8 +149,12 @@ call WHERE uvicorn > .tmp
for %%J in ("sd-v1-4.ckpt") do if "%%~zJ" EQU "7703807346" ( for %%J in ("sd-v1-4.ckpt") do if "%%~zJ" EQU "7703807346" (
echo "Data files (weights) necessary for Stable Diffusion were already downloaded" echo "Data files (weights) necessary for Stable Diffusion were already downloaded"
) else ( ) 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. for %%K in ("sd-v1-4.ckpt") do if "%%~zK" EQU "7703810927" (
del "sd-v1-4.ckpt" 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"
)
) )
) )
) )

View File

@ -162,7 +162,7 @@ if [ ! -f "sd-v1-4.ckpt" ]; then
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}'` 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: 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" 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" read -p "Press any key to continue"