BUG TypeError: GPT2Model.forward() got an unexpected keyword argument ‘past’

集电极2024-07-06 16:36

TypeError: GPT2Model.forward() got an unexpected keyword argument 'past'

环境

shell 复制代码

transformers                  4.38.1

详情

这是由于新版的transformers 对GPT2Model.forward() 参数进行了改变导致的错误。具体是past名称改为了 past_key_values 。

解决方法

找到错误语句，将其中的past 参数改为past_key_values 。

例如

python 复制代码

model = GPT2Model(config)
# 旧
transformer_outputs = model(input_ids, past=past, token_type_ids=token_type_ids)

# 新
transformer_outputs = model(input_ids, past_key_values=past, token_type_ids=token_type_ids)

参考

https://github.com/liucongg/GPT2-NewsTitle/issues/34