37 lines
1.2 KiB
Bash
Executable File
37 lines
1.2 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
set -e
|
|
|
|
declare -A MODELS
|
|
|
|
MODELS["${ROOT}/GFPGANv1.3.pth"]=GFPGANv1.3.pth
|
|
MODELS["${WORKDIR}/repositories/latent-diffusion/experiments/pretrained_models/model.chkpt"]=LDSR.ckpt
|
|
MODELS["${WORKDIR}/repositories/latent-diffusion/experiments/pretrained_models/project.yaml"]=LDSR.yaml
|
|
|
|
MODELS_DIR=/cache/models
|
|
|
|
for path in "${!MODELS[@]}"; do
|
|
name=${MODELS[$path]}
|
|
base=$(dirname "${path}")
|
|
from_path="${MODELS_DIR}/${name}"
|
|
if test -f "${from_path}"; then
|
|
mkdir -p "${base}" && ln -sf "${from_path}" "${path}" && echo "Mounted ${name}"
|
|
else
|
|
echo "Skipping ${name}"
|
|
fi
|
|
done
|
|
|
|
# force realesrgan cache
|
|
rm -rf /opt/conda/lib/python3.8/site-packages/realesrgan/weights
|
|
ln -s -T "${MODELS_DIR}" /opt/conda/lib/python3.8/site-packages/realesrgan/weights
|
|
|
|
# force facexlib cache
|
|
mkdir -p /cache/weights/ ${WORKDIR}/gfpgan/
|
|
ln -sf /cache/weights/ ${WORKDIR}/gfpgan/
|
|
# code former cache
|
|
rm -rf ${ROOT}/repositories/CodeFormer/weights/CodeFormer ${ROOT}/repositories/CodeFormer/weights/facelib
|
|
ln -sf -T /cache/weights ${ROOT}/repositories/CodeFormer/weights/CodeFormer
|
|
ln -sf -T /cache/weights ${ROOT}/repositories/CodeFormer/weights/facelib
|
|
|
|
mkdir -p /cache/torch /cache/transformers /cache/weights /cache/models /cache/custom-models
|