Skip to content

Commit 8e5cc7e

Browse files
committed
fix: check ``` in hf prefill
1 parent 2f9f4c8 commit 8e5cc7e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

bigcodebench/provider/vllm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ def __init__(self, name: str, dataset: str, tp: int, **kwargs) -> None:
2727
if self.is_direct_completion():
2828
self.eos += extra_eos_for_direct_completion(dataset)
2929
else:
30-
if self.prefill:
30+
if self.prefill and "```" in self.response_prefix:
3131
self.eos += ["\n```\n"]
3232
self.llm = LLM(model=name, max_model_len=self.max_new_tokens, **kwargs)
3333
self.llm.set_tokenizer(tokenizer=self.tokenizer)

0 commit comments

Comments
 (0)