2a0de025e2
As discussed in Discussion [#367](https://github.com/AbdBarho/stable-diffusion-webui-docker/discussions/367), this adds support for the newer ComfyUI. I forked the fork that would already add this but the maintainer of that fork hasn't implemented the changes needed to properly get the output function working, which I did. I believe everything is functional though I have not tested every single node. I changed the table format for the README and a few other minor things for aesthetic reasons but if you want me to revert those, I will. --------- Co-authored-by: Jonathan Kovacs <jkovacs-dev@users.noreply.github.com> Co-authored-by: AbdBarho <ka70911@gmail.com>
48 lines
1.4 KiB
Bash
48 lines
1.4 KiB
Bash
#!/bin/bash
|
|
|
|
set -Eeuo pipefail
|
|
|
|
declare -A MOUNTS
|
|
|
|
|
|
mkdir -vp /data/config/comfy/
|
|
|
|
# cache
|
|
MOUNTS["/root/.cache"]=/data/.cache
|
|
# ui specific
|
|
MOUNTS["${ROOT}/models/checkpoints"]="/data/StableDiffusion"
|
|
MOUNTS["${ROOT}/models/controlnet"]="/data/ControlNet"
|
|
MOUNTS["${ROOT}/models/upscale_models/RealESRGAN"]="/data/RealESRGAN"
|
|
MOUNTS["${ROOT}/models/upscale_models/GFPGAN"]="/data/GFPGAN"
|
|
MOUNTS["${ROOT}/models/upscale_models/SwinIR"]="/data/SwinIR"
|
|
MOUNTS["${ROOT}/models/vae"]="/data/VAE"
|
|
|
|
# data
|
|
MOUNTS["${ROOT}/models/loras"]="/data/Lora"
|
|
MOUNTS["${ROOT}/models/embeddings"]="/data/embeddings"
|
|
|
|
# config
|
|
# TODO: I am not sure if this is final, maybe it should change in the future
|
|
MOUNTS["${ROOT}/models/clip"]="/data/.cache/comfy/clip"
|
|
MOUNTS["${ROOT}/models/clip_vision"]="/data/.cache/comfy/clip_vision"
|
|
MOUNTS["${ROOT}/models/custom_nodes"]="/data/config/comfy/custom_nodes"
|
|
MOUNTS["${ROOT}/models/style_models"]="/data/config/comfy/style_models"
|
|
MOUNTS["${ROOT}/models/t2i_adapter"]="/data/config/comfy/t2i_adapter"
|
|
|
|
# output
|
|
MOUNTS["${ROOT}/output"]="/output/comfy"
|
|
|
|
for to_path in "${!MOUNTS[@]}"; do
|
|
set -Eeuo pipefail
|
|
from_path="${MOUNTS[${to_path}]}"
|
|
rm -rf "${to_path}"
|
|
if [ ! -f "$from_path" ]; then
|
|
mkdir -vp "$from_path"
|
|
fi
|
|
mkdir -vp "$(dirname "${to_path}")"
|
|
ln -sT "${from_path}" "${to_path}"
|
|
echo Mounted $(basename "${from_path}")
|
|
done
|
|
|
|
exec "$@"
|