William Gill
d935e24624
All checks were successful
release / Build & Push Docker Image (push) Successful in 44s
Five pieces:
1. Dockerfile installs storyicon/comfyui_segment_anything (GroundingDINO
+ SAM-HQ in one bundle) into custom_nodes and pip-installs its
requirements at build time. Model weights auto-download to the
comfyui-models volume on first inpaint (~3 GB one-time cost).
2. install-custom-node-deps.sh — entrypoint wrapper that pip-installs
requirements.txt for any custom_node present at startup. Lets users
add custom nodes via ComfyUI-Manager (or by git-cloning into the
volume) and have the deps picked up on the next restart, without
editing the Dockerfile.
3. smart_image_gen v0.6: edit_image gains a `mask_text` param. When
set, builds an inpainting workflow (LoadImage → GroundingDinoSAM
Segment → SetLatentNoiseMask → KSampler) so only the named region
is repainted. When unset, falls through to the existing img2img
path. Denoise default switches: 1.0 with mask_text (full repaint
within mask), 0.7 without.
4. Image Studio system prompt teaches the LLM the LOCAL vs GLOBAL
distinction — set mask_text whenever the user names a specific
object/region ('the ball', 'the dog', 'the sky'); leave it unset
only for whole-image style/lighting transformations.
5. Deployment README documents the new mode + the first-inpaint
weight-download caveat.
Image rebuild required — bump tag to pick up the Dockerfile change.
Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
76 lines
3.2 KiB
Docker
76 lines
3.2 KiB
Docker
# ComfyUI on NVIDIA — manual install per
|
|
# https://docs.comfy.org/installation/manual_install#nvidia
|
|
#
|
|
# Base image provides system CUDA / cuDNN libs that some custom nodes dlopen at
|
|
# runtime. The PyTorch wheels installed below bundle their own CUDA runtime —
|
|
# the base only needs to be ABI-compatible with the host driver via
|
|
# nvidia-container-toolkit.
|
|
#
|
|
# CUDA 12.6 + cu126 wheels chosen for broad driver compatibility (driver >=
|
|
# 545). If the host driver is >= 580, switch the wheel index URL to cu130 for
|
|
# the newest stable PyTorch.
|
|
|
|
FROM nvidia/cuda:12.6.3-cudnn-runtime-ubuntu24.04
|
|
|
|
ENV DEBIAN_FRONTEND=noninteractive \
|
|
PYTHONUNBUFFERED=1 \
|
|
PIP_NO_CACHE_DIR=1 \
|
|
PIP_DISABLE_PIP_VERSION_CHECK=1 \
|
|
COMFYUI_HOME=/opt/comfyui
|
|
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
python3.12 \
|
|
python3.12-venv \
|
|
python3-pip \
|
|
git \
|
|
ca-certificates \
|
|
curl \
|
|
libgl1 \
|
|
libglib2.0-0 \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
# Pin a venv so future pip installs (custom nodes) stay isolated.
|
|
RUN python3.12 -m venv /opt/venv
|
|
ENV PATH=/opt/venv/bin:$PATH
|
|
|
|
# Upstream wheel for NVIDIA. cu126 covers driver >= 545; bump to cu130 for
|
|
# driver >= 580.
|
|
RUN pip install --upgrade pip && \
|
|
pip install torch torchvision torchaudio \
|
|
--extra-index-url https://download.pytorch.org/whl/cu126
|
|
|
|
# Pull ComfyUI itself. Pinning to a tag would be safer for reproducibility but
|
|
# the project ships breaking changes infrequently and most users want latest.
|
|
RUN git clone --depth 1 https://github.com/comfyanonymous/ComfyUI.git ${COMFYUI_HOME}
|
|
WORKDIR ${COMFYUI_HOME}
|
|
RUN pip install -r requirements.txt
|
|
|
|
# ComfyUI-Manager — community node manager. Lets users install/update custom
|
|
# nodes from the web UI instead of editing the Dockerfile. Remove this block if
|
|
# you want a pristine ComfyUI.
|
|
RUN git clone --depth 1 https://github.com/ltdrdata/ComfyUI-Manager.git \
|
|
${COMFYUI_HOME}/custom_nodes/ComfyUI-Manager && \
|
|
pip install -r ${COMFYUI_HOME}/custom_nodes/ComfyUI-Manager/requirements.txt
|
|
|
|
# comfyui_segment_anything — GroundingDINO + SAM-HQ in one bundle. Required
|
|
# by the smart_image_gen Tool's text-targeted inpainting (edit_image with the
|
|
# mask_text parameter). Model weights auto-download on first use into
|
|
# /opt/comfyui/models/{sams,grounding-dino}/ — first inpaint takes ~3 GB of
|
|
# downloads, subsequent runs are instant.
|
|
RUN git clone --depth 1 https://github.com/storyicon/comfyui_segment_anything.git \
|
|
${COMFYUI_HOME}/custom_nodes/comfyui_segment_anything && \
|
|
pip install -q -r ${COMFYUI_HOME}/custom_nodes/comfyui_segment_anything/requirements.txt
|
|
|
|
# Entrypoint wrapper — auto-installs requirements.txt for any custom_node
|
|
# present at startup (covers Manager-installed nodes and nodes cloned
|
|
# directly into the comfyui-custom-nodes volume).
|
|
COPY install-custom-node-deps.sh /usr/local/bin/install-custom-node-deps.sh
|
|
RUN chmod +x /usr/local/bin/install-custom-node-deps.sh
|
|
|
|
EXPOSE 8188
|
|
|
|
# --listen 0.0.0.0 binds to every interface so the Open WebUI container on the
|
|
# shared compose network can reach it. --port is explicit for clarity.
|
|
ENTRYPOINT ["/usr/local/bin/install-custom-node-deps.sh"]
|
|
CMD ["python", "main.py", "--listen", "0.0.0.0", "--port", "8188"]
|