Pin transformers <5 — comfyui_segment_anything's GroundingDINO needs it
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>
This commit is contained in:
2026-04-19 15:21:21 -05:00
parent f26dfbee02
commit 2cecf77981
2 changed files with 14 additions and 1 deletions

View File

@@ -57,9 +57,15 @@ RUN git clone --depth 1 https://github.com/ltdrdata/ComfyUI-Manager.git \
# 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 -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

View File

@@ -18,4 +18,11 @@ if [ -d /opt/comfyui/custom_nodes ]; then
done
fi
# Force-pin known-incompatible packages back into a working range. Some
# custom nodes bring transformers >=5 transitively, which removes
# BertModel.get_head_mask and breaks comfyui_segment_anything's
# GroundingDINO. Run last so it wins over anything the loop above
# installed.
pip install -q "transformers>=4.40,<5" || echo "[entrypoint] transformers pin failed — continuing"
exec "$@"