mount.sh 1.0 KB

123456789101112131415161718192021222324252627282930313233343536
  1. #!/bin/bash
  2. set -Eeuo pipefail
  3. declare -A MOUNTS
  4. MOUNTS["/root/.cache"]="/data/.cache"
  5. # main
  6. MOUNTS["${ROOT}/models/Stable-diffusion"]="/data/StableDiffusion"
  7. MOUNTS["${ROOT}/models/Codeformer"]="/data/Codeformer"
  8. MOUNTS["${ROOT}/models/GFPGAN"]="/data/GFPGAN"
  9. MOUNTS["${ROOT}/models/ESRGAN"]="/data/ESRGAN"
  10. MOUNTS["${ROOT}/models/BSRGAN"]="/data/BSRGAN"
  11. MOUNTS["${ROOT}/models/RealESRGAN"]="/data/RealESRGAN"
  12. MOUNTS["${ROOT}/models/SwinIR"]="/data/SwinIR"
  13. MOUNTS["${ROOT}/models/ScuNET"]="/data/ScuNET"
  14. MOUNTS["${ROOT}/models/LDSR"]="/data/LDSR"
  15. MOUNTS["${ROOT}/models/hypernetworks"]="/data/Hypernetworks"
  16. MOUNTS["${ROOT}/embeddings"]="/data/embeddings"
  17. # extra hacks
  18. MOUNTS["${ROOT}/repositories/CodeFormer/weights/facelib"]="/data/.cache"
  19. for to_path in "${!MOUNTS[@]}"; do
  20. set -Eeuo pipefail
  21. from_path="${MOUNTS[${to_path}]}"
  22. rm -rf "${to_path}"
  23. mkdir -vp "$from_path"
  24. mkdir -vp "$(dirname "${to_path}")"
  25. ln -sT "${from_path}" "${to_path}"
  26. echo Mounted $(basename "${from_path}")
  27. done
  28. mkdir -p /output/saved