当前位置: 首页 > news >正文

使用transformers过程中出现的bug

1. The following model_kwargs are not used by the model: ['encoder_hidden_states', 'encoder_attention_mask'] (note: typos in the generate arguments will also show up in this list)

使用text_decoder就出现上述错误,这是由于transformers版本不兼容导致的

from transformers import AutoModel, AutoConfig, BertGenerationDecoder
decoder_config = AutoConfig.from_pretrained(args['text_checkpoint'])

text_decoder = BertGenerationDecoder(config=decoder_config)

output = self.text_decoder.generate(input_ids=cls_input_ids, 
                                    encoder_hidden_states=encoder_hidden_states, 
                                    encoder_attention_mask=encoder_attention_mask,  
                                    max_length=self.args['max_seq_length'], 
                                    do_sample=True, 
                                    num_beams=self.args['beam_size'], 
                                    length_penalty=1.0, use_cache=True,
                                    )

解决办法:将transformer的版本换到以下范围, 4.15.0<=transformers<4.22.0,transformers>=4.25.0

比如:pip install transformers==4.25.1 or pip install transformers==4.20.1

相关文章:

  • notepad++ 批量替换删除指定字符之后 或者 之前的字符,Notepad+批量替换使用大全
  • vue3学习(七)--- Teleport传送组件
  • 使用Golang实现HTTP代理突破IP访问限制
  • react写一个简单的3d滚轮picker组件
  • postgresql|数据库|恢复备份的时候报错:pg_restore: implied data-only restore的处理方案
  • rabbitMQ的知识点
  • Java|学习|异常
  • Vue2 与 React 的区别
  • JOSEF约瑟 漏电继电器 JD1-200 工作电压:380V 孔径:45mm 50~500mA
  • js + selenium 获取chatgpt的accessToken
  • Java实现B树
  • 解决uniapp里scroll-view横向滚动的问题
  • JOSEF约瑟 多档切换式漏电(剩余)继电器JHOK-ZBL1 30/100/300/500mA
  • 【Redis】Java Spring操作redis
  • 2024上海国际智慧城市展览会(世亚智博会)智慧城市,数字中国
  • Python数据挖掘:入门、进阶与实用案例分析——自动售货机销售数据分析与应用
  • 淘宝天猫商品评论数据接口,淘宝天猫商品评论API接口,淘宝API
  • 【Rust基础①】基本类型、所有权与借用、复合类型
  • 【Vivado HLS Bug】Ubuntu环境下Vivado HLS导出IP报错:HLS ERROR: [IMPL 213-28]
  • Ant Design Vue Element-ui 中table 合并行功能,以及带分页并合并行
  • 绍兴柯桥:用一块布托起中国制造的新经纬
  • 南通市委常委、市委秘书长童剑跨市调任常州市委常委、组织部部长
  • 多地跟进官宣下调公积金贷款利率,最低降至2.1%
  • 印方称若巴方决定升级局势,印方已做好反击准备
  • 上海营商环境的“分寸”感:底线之上不断拓宽自由,底线之下雷霆制止
  • 两次蹚入同一条河,巴萨这一晚被命运抛弃