Skip to content

GitLab

Unverified Commit 5f5f35e5 authored by weiliang's avatar weiliang Committed by GitHub
Browse files

fix vllm test issue that call pop() from None (#3182)

parent 314f7176
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
lm_eval/models/vllm_causallms.py
View file @ 5f5f35e5
...@@ -209,7 +209,7 @@ class VLLM(TemplateLM): ...@@ -209,7 +209,7 @@ class VLLM(TemplateLM):
) )
self.tokenizer = configure_pad_token(self.tokenizer, model_config=self._config) self.tokenizer = configure_pad_token(self.tokenizer, model_config=self._config)
self.chat_template_args = chat_template_args or {} self.chat_template_args = chat_template_args or {}
self.enable_thinking = chat_template_args.pop( self.enable_thinking = self.chat_template_args.pop(
"enable_thinking", enable_thinking "enable_thinking", enable_thinking
) )
self.add_bos_token = add_bos_token self.add_bos_token = add_bos_token
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment