stable-diffusion-webui-docker/services/AUTOMATIC1111/mount.sh

32 lines
878 B
Bash
Raw Normal View History

#!/bin/bash
2022-10-01 10:53:48 +02:00
set -Eeuo pipefail
2022-10-01 12:17:39 +02:00
declare -A MOUNTS
2022-10-01 10:53:48 +02:00
2022-10-01 12:17:39 +02:00
MOUNTS["/root/.cache"]="/data/.cache"
2022-10-01 10:53:48 +02:00
# main
2022-10-01 12:17:39 +02:00
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"
2022-10-01 10:53:48 +02:00
2022-10-01 12:27:24 +02:00
MOUNTS["${ROOT}/embeddings"]="/data/embeddings"
2022-10-01 10:53:48 +02:00
# extra hacks
2022-10-01 12:17:39 +02:00
MOUNTS["${ROOT}/repositories/CodeFormer/weights/facelib"]="/data/.cache"
2022-10-01 10:53:48 +02:00
2022-10-01 12:17:39 +02:00
for to_path in "${!MOUNTS[@]}"; do
2022-10-01 10:53:48 +02:00
set -Eeuo pipefail
2022-10-01 12:17:39 +02:00
from_path="${MOUNTS[${to_path}]}"
2022-10-01 10:53:48 +02:00
rm -rf "${to_path}"
mkdir -p "$(dirname "${to_path}")"
ln -sT "${from_path}" "${to_path}"
echo Mounted $(basename "${from_path}")
done