We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2f9f4c8 commit 8e5cc7eCopy full SHA for 8e5cc7e
bigcodebench/provider/vllm.py
@@ -27,7 +27,7 @@ def __init__(self, name: str, dataset: str, tp: int, **kwargs) -> None:
27
if self.is_direct_completion():
28
self.eos += extra_eos_for_direct_completion(dataset)
29
else:
30
- if self.prefill:
+ if self.prefill and "```" in self.response_prefix:
31
self.eos += ["\n```\n"]
32
self.llm = LLM(model=name, max_model_len=self.max_new_tokens, **kwargs)
33
self.llm.set_tokenizer(tokenizer=self.tokenizer)
0 commit comments