'BertLMHeadModel' object has no attribute 'generate'

[yesokyin]

Traceback (most recent call last):
File "/Grounded-Segment-Anything-main/automatic_label_tag2text_demo.py", line 306, in
res = inference_tag2text(raw_image , tag2text_model, specified_tags)
File "/Grounded-Segment-Anything-main/recognize-anything/ram/inference.py", line 11, in inference_tag2text
caption, tag_predict = model.generate(image,
File "/Grounded-Segment-Anything-main/recognize-anything/ram/models/tag2text.py", line 355, in generate
outputs = self.text_decoder.generate(
File "/granded_sam/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1931, in getattr
raise AttributeError(
AttributeError: 'BertLMHeadModel' object has no attribute 'generate'

[RobinChen007]

It seems that there is a problem with the version of torch

[Sinabroslowly]

I am currently having the same issue while executing automatic_label_tag2text_demo.py

[NiuCY]

pip install transformers==4.41.2, it work.