From 743f60577364d43237eeb46aefd523371c36e56b Mon Sep 17 00:00:00 2001 From: mauwii Date: Tue, 1 Nov 2022 01:21:15 +0100 Subject: [PATCH] update build.sh to download sd-v1.5 model --- docker-build/build.sh | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/docker-build/build.sh b/docker-build/build.sh index fea311ae82..a0e97ca87f 100755 --- a/docker-build/build.sh +++ b/docker-build/build.sh @@ -38,11 +38,12 @@ _copyCheckpoints() { echo "creating subfolders for models and outputs" _runAlpine mkdir models _runAlpine mkdir outputs - echo -n "downloading sd-v1-4.ckpt" - _runAlpine wget --header="Authorization: Bearer ${huggingface_token}" -O models/sd-v1-4.ckpt https://huggingface.co/CompVis/stable-diffusion-v-1-4-original/resolve/main/sd-v1-4.ckpt + echo "downloading v1-5-pruned-emaonly.ckpt" + _runAlpine wget \ + --header="Authorization: Bearer ${huggingface_token}" \ + -O models/v1-5-pruned-emaonly.ckpt \ + https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.ckpt echo "done" - echo "downloading GFPGANv1.4.pth" - _runAlpine wget -O models/GFPGANv1.4.pth https://github.com/TencentARC/GFPGAN/releases/download/v1.3.0/GFPGANv1.4.pth } _checkVolumeContent() { @@ -51,7 +52,7 @@ _checkVolumeContent() { _getModelMd5s() { _runAlpine \ - alpine sh -c "md5sum /data/models/*" + alpine sh -c "md5sum /data/models/*.ckpt" } if [[ -n "$(docker volume ls -f name="${volumename}" -q)" ]]; then