Skip to content

AttributeError: 'BertModel' object has no attribute 'get_head_mask' #140

Description

@shsmad
  File "/stablediff/StabilityMatrix/Packages/ComfyUI/custom_nodes/ComfyUI_LayerStyle_Advance/py/segment_anything_ultra.py", line 48, in segment_anything_ultra
    self.DINO_MODEL = load_groundingdino_model(grounding_dino_model)
                      ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/stablediff/StabilityMatrix/Packages/ComfyUI/custom_nodes/ComfyUI_LayerStyle_Advance/py/segment_anything_func.py", line 121, in load_groundingdino_model
    dino = local_groundingdino_build_model(dino_model_args)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/stablediff/StabilityMatrix/Packages/ComfyUI/custom_nodes/comfyui_layerstyle/py/local_groundingdino/models/__init__.py", line 17, in build_model
    model = build_func(args)
            ^^^^^^^^^^^^^^^^
  File "/stablediff/StabilityMatrix/Packages/ComfyUI/custom_nodes/comfyui_layerstyle/py/local_groundingdino/models/GroundingDINO/groundingdino.py", line 362, in build_groundingdino
    model = GroundingDINO(
            ^^^^^^^^^^^^^^
  File "/stablediff/StabilityMatrix/Packages/ComfyUI/custom_nodes/comfyui_layerstyle/py/local_groundingdino/models/GroundingDINO/groundingdino.py", line 101, in __init__
    self.bert = BertModelWarper(bert_model=self.bert)
                ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/stablediff/StabilityMatrix/Packages/ComfyUI/custom_nodes/comfyui_layerstyle/py/local_groundingdino/models/GroundingDINO/bertwarper.py", line 25, in __init__
    self.get_head_mask = bert_model.get_head_mask
                         ^^^^^^^^^^^^^^^^^^^^^^^^
  File "/stablediff/StabilityMatrix/Packages/ComfyUI/venv/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1965, in __getattr__
    raise AttributeError(
AttributeError: 'BertModel' object has no attribute 'get_head_mask'

transformers 5.2.0
peft 0.18.1
torch 2.10.0+cu130
ComfyUI_LayerStyle 2.0.38
ComfyUI_LayerStyle_Advance 2.0.37
setuptools 81.0.0
ubuntu 24.04
NVIDIA GeForce RTX 5060 Ti

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions