Fix revision parameter to vllm get_tokenizer (#2492)

EleutherAI · Nov 15, 2024 · e20e1dd · e20e1dd
1 parent 4259a6d
commit e20e1dd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lm_eval/models/vllm_causallms.py b/lm_eval/models/vllm_causallms.py
@@ -118,7 +118,7 @@ def __init__(
             tokenizer if tokenizer else pretrained,
             tokenizer_mode=tokenizer_mode,
             trust_remote_code=trust_remote_code,
-            tokenizer_revision=tokenizer_revision,
+            revision=tokenizer_revision,
         )
         self.tokenizer = configure_pad_token(self.tokenizer)
         self.add_bos_token = add_bos_token