NotImplementedError: Make sure that a _reorder_cache function is correctly implemented

[dmyshkin]

Hello,

I followed the quick start collab to run AraGPT2 generation:
https://colab.research.google.com/drive/1T5xKRAtKeuUmmokyThvkK8pulHw02CoK?usp=sharing#scrollTo=tplhJoCTkjci

All of the commands worked until it came time to call :
gen_text = aragpt2_pipeline(text_prep,
pad_token_id=0, # 0 for AraGPT2
num_beams=5,
max_length=200,
top_p=0.95,
repetition_penalty = 3.0,
no_repeat_ngram_size = 3)[0]['generated_text']

I got the following error:
NotImplementedError: Make sure that a _reorder_cache function is correctly implemented in arabert.aragpt2.grover.modeling_gpt2 to enable beam search for <class 'arabert.aragpt2.grover.modeling_gpt2.GPT2LMHeadModel'>

Any ideas on how to fix this would be appreciated.

[dmyshkin]

Nevermind, I think I was using too new of a transformers version for that how-to. Was using transformers 4.7.0 and the how-to has it doing transformers 4.2.1