Fix GenerationMixin inheritance for transformers >= 4.50
#3
by roemmele - opened
RNNLMForCausalLM now explicitly inherits from GenerationMixin (in addition to PreTrainedModel), fixing an AttributeError on _extract_generation_mode_kwargs when calling generate() with transformers >= 4.50.
One-line change: class RNNLMForCausalLM(PreTrainedModel, GenerationMixin):
roemmele changed pull request status to merged
roemmele deleted the
refs/pr/3 ref