10c16e1971
Fixes a problem with cross attention class missing from diffusers
models are now taken from the huggingFace cache.
50eb02f68b
48 lines
1.0 KiB
Bash
Executable File
48 lines
1.0 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
set -Eeuo pipefail
|
|
|
|
declare -A MOUNTS
|
|
|
|
mkdir -p ${CONFIG_DIR}
|
|
|
|
# cache
|
|
MOUNTS["/root/.cache"]=/data/.cache/
|
|
|
|
# this is really just a hack to avoid migrations
|
|
rm -rf ${HF_HOME}/diffusers
|
|
|
|
# ui specific
|
|
MOUNTS["${ROOT}/models/codeformer"]=/data/Codeformer/
|
|
MOUNTS["${ROOT}/models/gfpgan/GFPGANv1.4.pth"]=/data/GFPGAN/GFPGANv1.4.pth
|
|
MOUNTS["${ROOT}/models/gfpgan/weights"]=/data/GFPGAN/
|
|
MOUNTS["${ROOT}/models/realesrgan"]=/data/RealESRGAN/
|
|
|
|
MOUNTS["${ROOT}/models/ldm"]=/data/.cache/invoke/ldm/
|
|
|
|
MOUNTS["${ROOT}/embeddings"]=/data/embeddings/
|
|
|
|
# hacks
|
|
|
|
for to_path in "${!MOUNTS[@]}"; do
|
|
set -Eeuo pipefail
|
|
from_path="${MOUNTS[${to_path}]}"
|
|
rm -rf "${to_path}"
|
|
mkdir -p "$(dirname "${to_path}")"
|
|
# ends with slash, make it!
|
|
if [[ "$from_path" == */ ]]; then
|
|
mkdir -vp "$from_path"
|
|
fi
|
|
|
|
ln -sT "${from_path}" "${to_path}"
|
|
echo Mounted $(basename "${from_path}")
|
|
done
|
|
|
|
if "${PRELOAD}" == "true"; then
|
|
set -Eeuo pipefail
|
|
invokeai-configure --root ${ROOT} --yes
|
|
cp ${ROOT}/configs/models.yaml ${CONFIG_DIR}/models.yaml
|
|
fi
|
|
|
|
exec "$@"
|