You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
33 lines
840 B
Bash
33 lines
840 B
Bash
#!/bin/bash
|
|
|
|
set -Eeuo pipefail
|
|
|
|
declare -A MOUNTS
|
|
|
|
ROOT=/stable-diffusion/src
|
|
|
|
# cache
|
|
MOUNTS["/root/.cache"]=/data/.cache
|
|
# ui specific
|
|
MOUNTS["${PWD}/models/realesrgan"]=/data/RealESRGAN
|
|
MOUNTS["${PWD}/models/ldsr"]=/data/LDSR
|
|
MOUNTS["${PWD}/models/custom"]=/data/StableDiffusion
|
|
|
|
# hack
|
|
MOUNTS["${PWD}/models/gfpgan/GFPGANv1.3.pth"]=/data/GFPGAN/GFPGANv1.4.pth
|
|
MOUNTS["${PWD}/models/gfpgan/GFPGANv1.4.pth"]=/data/GFPGAN/GFPGANv1.4.pth
|
|
MOUNTS["${PWD}/gfpgan/weights"]=/data/.cache
|
|
|
|
|
|
for to_path in "${!MOUNTS[@]}"; do
|
|
set -Eeuo pipefail
|
|
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
|
|
|
|
# streamlit config
|
|
ln -sf /docker/userconfig_streamlit.yaml /stable-diffusion/configs/webui/userconfig_streamlit.yaml
|