| 1234567891011121314151617181920212223242526272829303132 |
- #!/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
|