| 123456789101112131415161718192021222324252627282930313233343536 |
- #!/bin/bash
- set -Eeuo pipefail
- declare -A MOUNTS
- MOUNTS["/root/.cache"]="/data/.cache"
- # main
- 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/ScuNET"]="/data/ScuNET"
- MOUNTS["${ROOT}/models/LDSR"]="/data/LDSR"
- MOUNTS["${ROOT}/models/hypernetworks"]="/data/Hypernetworks"
- MOUNTS["${ROOT}/embeddings"]="/data/embeddings"
- # extra hacks
- MOUNTS["${ROOT}/repositories/CodeFormer/weights/facelib"]="/data/.cache"
- for to_path in "${!MOUNTS[@]}"; do
- set -Eeuo pipefail
- from_path="${MOUNTS[${to_path}]}"
- rm -rf "${to_path}"
- mkdir -vp "$from_path"
- mkdir -vp "$(dirname "${to_path}")"
- ln -sT "${from_path}" "${to_path}"
- echo Mounted $(basename "${from_path}")
- done
- mkdir -p /output/saved
|