diff --git a/services/AUTOMATIC1111/mount.sh b/services/AUTOMATIC1111/mount.sh index 13f4f4a..3f8deb5 100755 --- a/services/AUTOMATIC1111/mount.sh +++ b/services/AUTOMATIC1111/mount.sh @@ -2,60 +2,28 @@ set -Eeuo pipefail -rm -rf /root/.cache -ln -sT /data/.cache /root/.cache +declare -A MOUNTS -declare -A FOLDERS +MOUNTS["/root/.cache"]="/data/.cache" # main -FOLDERS["/data/StableDiffusion"]="${ROOT}/models/Stable-diffusion" -FOLDERS["/data/Codeformer"]="${ROOT}/models/Codeformer" -FOLDERS["/data/GFPGAN"]="${ROOT}/models/GFPGAN" -FOLDERS["/data/ESRGAN"]="${ROOT}/models/ESRGAN" -FOLDERS["/data/BSRGAN"]="${ROOT}/models/BSRGAN" -FOLDERS["/data/RealESRGAN"]="${ROOT}/models/RealESRGAN" -FOLDERS["/data/SwinIR"]="${ROOT}/models/SwinIR" -FOLDERS["/data/LDSR"]="${ROOT}/models/LDSR" +MOUNTS["${ROOT}/models/Stable-diffusion"]="/data/StableDiffusion" +MOUNTS["${ROOT}/models/Codeformer"]="/data/Codeformer" +MOUNTS["${ROOT}/models/GFPGAN"]="/data/GFPGAN" +MOUNTS["${ROOT}/models/ESRGAN"]="/data/ESRGAN" +MOUNTS["${ROOT}/models/BSRGAN"]="/data/BSRGAN" +MOUNTS["${ROOT}/models/RealESRGAN"]="/data/RealESRGAN" +MOUNTS["${ROOT}/models/SwinIR"]="/data/SwinIR" +MOUNTS["${ROOT}/models/LDSR"]="/data/LDSR" # extra hacks -FOLDERS["/data/.cache"]="${ROOT}/repositories/CodeFormer/weights/facelib" +MOUNTS["${ROOT}/repositories/CodeFormer/weights/facelib"]="/data/.cache" -for from_path in "${!FOLDERS[@]}"; do +for to_path in "${!MOUNTS[@]}"; do set -Eeuo pipefail - to_path="${FOLDERS[${from_path}]}" + from_path="${MOUNTS[${to_path}]}" rm -rf "${to_path}" mkdir -p "$(dirname "${to_path}")" ln -sT "${from_path}" "${to_path}" echo Mounted $(basename "${from_path}") done - -# declare -A MODELS - -# 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 -# name=${MODELS[$path]} -# base=$(dirname "${path}") -# from_path="${MODELS_DIR}/${name}" -# if test -f "${from_path}"; then -# mkdir -p "${base}" && ln -sf "${from_path}" "${path}" && echo "Mounted ${name}" -# else -# echo "Skipping ${name}" -# fi -# done - -# # force realesrgan cache -# rm -rf /opt/conda/lib/python3.8/site-packages/realesrgan/weights -# ln -s -T "${MODELS_DIR}" /opt/conda/lib/python3.8/site-packages/realesrgan/weights - -# # force facexlib cache -# mkdir -p /cache/weights/ ${WORKDIR}/gfpgan/ -# ln -sf /cache/weights/ ${WORKDIR}/gfpgan/ -# # code former cache -# rm -rf ${ROOT}/repositories/CodeFormer/weights/CodeFormer ${ROOT}/repositories/CodeFormer/weights/facelib -# ln -sf -T /cache/weights ${ROOT}/repositories/CodeFormer/weights/CodeFormer -# ln -sf -T /cache/weights ${ROOT}/repositories/CodeFormer/weights/facelib