From 15a61a99d663937822237ac88881614a72aa9c78 Mon Sep 17 00:00:00 2001 From: AbdBarho Date: Fri, 23 Sep 2022 16:38:50 +0200 Subject: [PATCH] Explicit path to GFPGAN model (#91) Refs #89 --- services/AUTOMATIC1111/Dockerfile | 2 +- services/AUTOMATIC1111/mount.sh | 2 -- 2 files changed, 1 insertion(+), 3 deletions(-) diff --git a/services/AUTOMATIC1111/Dockerfile b/services/AUTOMATIC1111/Dockerfile index 1670c1e..0ebc9aa 100644 --- a/services/AUTOMATIC1111/Dockerfile +++ b/services/AUTOMATIC1111/Dockerfile @@ -68,4 +68,4 @@ WORKDIR ${WORKDIR} EXPOSE 7860 # run, -u to not buffer stdout / stderr CMD /docker/mount.sh && \ - python3 -u ../../webui.py --listen --port 7860 --hide-ui-dir-config --ckpt-dir /cache/custom-models --ckpt /cache/models/model.ckpt ${CLI_ARGS} + python3 -u ../../webui.py --listen --port 7860 --hide-ui-dir-config --ckpt-dir /cache/custom-models --ckpt /cache/models/model.ckpt --gfpgan-model /cache/models/GFPGANv1.3.pth ${CLI_ARGS} diff --git a/services/AUTOMATIC1111/mount.sh b/services/AUTOMATIC1111/mount.sh index 5083a54..671edc6 100755 --- a/services/AUTOMATIC1111/mount.sh +++ b/services/AUTOMATIC1111/mount.sh @@ -4,12 +4,10 @@ set -e declare -A MODELS -MODELS["${WORKDIR}/models/ldm/stable-diffusion-v1/model.ckpt"]=model.ckpt MODELS["${ROOT}/GFPGANv1.3.pth"]=GFPGANv1.3.pth MODELS["${WORKDIR}/repositories/latent-diffusion/experiments/pretrained_models/model.chkpt"]=LDSR.ckpt MODELS["${WORKDIR}/repositories/latent-diffusion/experiments/pretrained_models/project.yaml"]=LDSR.yaml - MODELS_DIR=/cache/models for path in "${!MODELS[@]}"; do