Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
FIX Prompt learning with latest transformers error (#2140)
The error in PEFT is occurring after this transformers change: huggingface/transformers#33870 Now, in our tests, some model_kwargs no longer necessarily contain past_key_values, resulting in a KeyError. We now account for this possibility. Affected models were opt and gpt2.
- Loading branch information