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)