feat:添加sd-comfyui到列表

apps/sd-comfyui/latest/comfy/Dockerfile

@@ -0,0 +1,32 @@
+FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
+
+ENV DEBIAN_FRONTEND=noninteractive
+ENV PIP_PREFER_BINARY=1
+ENV ROOT=/apt
+
+RUN mkdir ${ROOT}
+WORKDIR ${ROOT}
+
+# Install Essential Packages (APT)
+RUN apt-get update && apt-get install -y git python3 python3-pip && apt-get clean
+
+# Get Comfy
+RUN git clone https://github.com/comfyanonymous/ComfyUI.git .
+
+# Install Essential Packages (PIP)
+RUN --mount=type=cache,target=/root/.cache/pip \
+  pip install torch torchvision --index-url https://download.pytorch.org/whl/cu118 && \
+  pip install -r requirements.txt && \
+  pip install --pre xformers && \
+  pip install triton
+
+# Setting up
+COPY entrypoint.sh /docker/entrypoint.sh
+RUN chmod +x /docker/entrypoint.sh
+ENTRYPOINT ["/docker/entrypoint.sh"]
+
+ENV PYTHONPATH="${PYTHONPATH}:${PWD}"
+
+CMD test -z "${APT_ARGS}" || apt-get -y install ${APT_ARGS} \
+    && test -z "${PIP_ARGS}" || pip install ${PIP_ARGS} \
+    && python3 -u main.py ${CLI_ARGS}

apps/sd-comfyui/latest/comfy/entrypoint.sh

@@ -0,0 +1,46 @@
+#!/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 "$@"