diff --git a/lm_eval/models/gpt2.py b/lm_eval/models/gpt2.py index c69b614255..caae4ad611 100644 --- a/lm_eval/models/gpt2.py +++ b/lm_eval/models/gpt2.py @@ -19,6 +19,7 @@ def __init__( load_in_8bit: Optional[bool] = False, trust_remote_code: Optional[bool] = False, use_fast: Optional[bool] = True, + additional_special_tokens: Optional[str] = None, ): super().__init__() @@ -64,6 +65,7 @@ def __init__( revision=revision, trust_remote_code=trust_remote_code, use_fast=use_fast, + additional_special_tokens=additional_special_tokens, ) self.vocab_size = self.tokenizer.vocab_size