William Gill
2cecf77981
All checks were successful
release / Build & Push Docker Image (push) Successful in 1m12s
transformers 5.0 removed BertModel.get_head_mask (it was on the legacy
4.x API). comfyui_segment_anything's GroundingDINO bertwarper.py still
calls bert_model.get_head_mask in __init__, so first inpaint crashes
with AttributeError. Pinned transformers>=4.40,<5 in two places:
- Dockerfile: applied AFTER the custom node's requirements.txt
install so it wins on a fresh image build.
- install-custom-node-deps.sh entrypoint: re-applied at every
container start so any future custom-node install (via
ComfyUI-Manager or volume clone) that pulls a newer transformers
transitively gets pinned back into the working range.
Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
82 lines
3.5 KiB
Docker
82 lines
3.5 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.
|
|
#
|
|
# Transformers must stay <5: GroundingDINO inside this node calls
|
|
# BertModel.get_head_mask, which transformers 5.0 silently removed. The pin
|
|
# is applied AFTER the requirements install so it overrides anything the
|
|
# upstream requirements.txt would have pulled.
|
|
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 && \
|
|
pip install -q "transformers>=4.40,<5"
|
|
|
|
# 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"]
|