From cd00cd4f04f9177e7ac165e2f835f922fd693eb6 Mon Sep 17 00:00:00 2001 From: zohair-fword Date: Wed, 18 Sep 2024 00:14:57 -0400 Subject: [PATCH] Bug Fixes and changes added changes to dockerfile and auth token from huggingface to download alternate files --- services/AUTOMATIC1111/Dockerfile | 5 +++-- services/download/download.sh | 2 +- services/download/links.txt | 6 ++++-- 3 files changed, 8 insertions(+), 5 deletions(-) diff --git a/services/AUTOMATIC1111/Dockerfile b/services/AUTOMATIC1111/Dockerfile index d595784..6e49ff7 100644 --- a/services/AUTOMATIC1111/Dockerfile +++ b/services/AUTOMATIC1111/Dockerfile @@ -1,4 +1,4 @@ -FROM alpine/git:2.36.2 as download +FROM alpine/git:2.36.2 AS download COPY clone.sh /clone.sh @@ -31,7 +31,8 @@ RUN --mount=type=cache,target=/root/.cache/pip \ git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git && \ cd stable-diffusion-webui && \ git reset --hard v1.9.4 && \ - pip install -r requirements_versions.txt + pip install -r requirements_versions.txt && \ + pip install -U typing_extensions ENV ROOT=/stable-diffusion-webui diff --git a/services/download/download.sh b/services/download/download.sh index fbe5a70..7006b48 100755 --- a/services/download/download.sh +++ b/services/download/download.sh @@ -15,7 +15,7 @@ mkdir -vp /data/.cache \ echo "Downloading, this might take a while..." -aria2c -x 10 --disable-ipv6 --input-file /docker/links.txt --dir /data/models --continue +aria2c --header "Authorization: Bearer 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" -x 10 --disable-ipv6 --input-file /docker/links.txt --dir /data/models --continue echo "Checking SHAs..." diff --git a/services/download/links.txt b/services/download/links.txt index 3616449..0953e59 100644 --- a/services/download/links.txt +++ b/services/download/links.txt @@ -1,8 +1,10 @@ -https://huggingface.co/runwayml/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.ckpt +https://huggingface.co/benjamin-paine/stable-diffusion-v1-5/resolve/main/v1-5-pruned-emaonly.ckpt out=Stable-diffusion/v1-5-pruned-emaonly.ckpt +https://huggingface.co/RunDiffusion/Juggernaut-XL-v9/resolve/main/Juggernaut-XL_v9_RunDiffusionPhoto_v2.safetensors?download=true + out=Stable-diffusion/Juggernaut-XL_v9_RunDiffusionPhoto_v2.safetensors https://huggingface.co/stabilityai/sd-vae-ft-mse-original/resolve/main/vae-ft-mse-840000-ema-pruned.ckpt out=VAE/vae-ft-mse-840000-ema-pruned.ckpt -https://huggingface.co/runwayml/stable-diffusion-inpainting/resolve/main/sd-v1-5-inpainting.ckpt +https://huggingface.co/benjamin-paine/stable-diffusion-v1-5-inpainting/resolve/main/sd-v1-5-inpainting.ckpt out=Stable-diffusion/sd-v1-5-inpainting.ckpt https://github.com/TencentARC/GFPGAN/releases/download/v1.3.4/GFPGANv1.4.pth out=GFPGAN/GFPGANv1.4.pth