AbdBarho/stable-diffusion-webui-docker
1
0
Fork 0
mirror of https://github.com/AbdBarho/stable-diffusion-webui-docker.git synced 2026-02-07 16:14:18 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

revert mount order

Browse source
This commit is contained in:
AbdBarho 2022-10-01 12:17:39 +02:00
parent 5b547a515d
commit 0f6fb2e67b
1 changed files with 13 additions and 45 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

58
services/AUTOMATIC1111/mount.sh
View file

@ -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