mirror of
https://github.com/AbdBarho/stable-diffusion-webui-docker.git
synced 2025-10-27 08:14:26 -04:00
@@ -19,30 +19,14 @@ if [ ! -f /data/config/auto/styles.csv ]; then
|
||||
touch /data/config/auto/styles.csv
|
||||
fi
|
||||
|
||||
# copy models from original models folder
|
||||
rsync -a --info=NAME ${ROOT}/models/VAE-approx/ /data/models/VAE-approx/
|
||||
rsync -a --info=NAME ${ROOT}/models/karlo/ /data/models/karlo/
|
||||
|
||||
declare -A MOUNTS
|
||||
|
||||
MOUNTS["/root/.cache"]="/data/.cache"
|
||||
|
||||
# main
|
||||
MOUNTS["${ROOT}/models/Stable-diffusion"]="/data/StableDiffusion"
|
||||
MOUNTS["${ROOT}/models/VAE"]="/data/VAE"
|
||||
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}/models/torch_deepdanbooru"]="/data/Deepdanbooru"
|
||||
MOUNTS["${ROOT}/models/BLIP"]="/data/BLIP"
|
||||
MOUNTS["${ROOT}/models/midas"]="/data/MiDaS"
|
||||
MOUNTS["${ROOT}/models/Lora"]="/data/Lora"
|
||||
MOUNTS["${ROOT}/models/LyCORIS"]="/data/LyCORIS"
|
||||
MOUNTS["${ROOT}/models/ControlNet"]="/data/ControlNet"
|
||||
MOUNTS["${ROOT}/models/openpose"]="/data/openpose"
|
||||
MOUNTS["${ROOT}/models/ModelScope"]="/data/ModelScope"
|
||||
MOUNTS["${ROOT}/models"]="/data/models"
|
||||
|
||||
MOUNTS["${ROOT}/embeddings"]="/data/embeddings"
|
||||
MOUNTS["${ROOT}/config.json"]="/data/config/auto/config.json"
|
||||
@@ -66,8 +50,21 @@ for to_path in "${!MOUNTS[@]}"; do
|
||||
echo Mounted $(basename "${from_path}")
|
||||
done
|
||||
|
||||
echo "Installing extension dependencies (if any)"
|
||||
|
||||
# because we build our container as root:
|
||||
chown -R root ~/.cache/
|
||||
chmod 766 ~/.cache/
|
||||
|
||||
shopt -s nullglob
|
||||
list=(./extensions/*/requirements.txt)
|
||||
for req in "${list[@]}"; do
|
||||
pip install -r "$req"
|
||||
done
|
||||
|
||||
if [ -f "/data/config/auto/startup.sh" ]; then
|
||||
pushd ${ROOT}
|
||||
echo "Running startup script"
|
||||
. /data/config/auto/startup.sh
|
||||
popd
|
||||
fi
|
||||
|
||||
Reference in New Issue
Block a user