mount.sh 840 B

1234567891011121314151617181920212223242526272829303132
  1. #!/bin/bash
  2. set -Eeuo pipefail
  3. declare -A MOUNTS
  4. ROOT=/stable-diffusion/src
  5. # cache
  6. MOUNTS["/root/.cache"]=/data/.cache
  7. # ui specific
  8. MOUNTS["${PWD}/models/realesrgan"]=/data/RealESRGAN
  9. MOUNTS["${PWD}/models/ldsr"]=/data/LDSR
  10. MOUNTS["${PWD}/models/custom"]=/data/StableDiffusion
  11. # hack
  12. MOUNTS["${PWD}/models/gfpgan/GFPGANv1.3.pth"]=/data/GFPGAN/GFPGANv1.4.pth
  13. MOUNTS["${PWD}/models/gfpgan/GFPGANv1.4.pth"]=/data/GFPGAN/GFPGANv1.4.pth
  14. MOUNTS["${PWD}/gfpgan/weights"]=/data/.cache
  15. for to_path in "${!MOUNTS[@]}"; do
  16. set -Eeuo pipefail
  17. from_path="${MOUNTS[${to_path}]}"
  18. rm -rf "${to_path}"
  19. mkdir -p "$(dirname "${to_path}")"
  20. ln -sT "${from_path}" "${to_path}"
  21. echo Mounted $(basename "${from_path}")
  22. done
  23. # streamlit config
  24. ln -sf /docker/userconfig_streamlit.yaml /stable-diffusion/configs/webui/userconfig_streamlit.yaml