From dcff46f86eac3b2da13ab519fb60308e3849a1b2 Mon Sep 17 00:00:00 2001 From: Terry Zhuo Date: Sat, 1 Feb 2025 09:13:42 +0800 Subject: [PATCH] fix prefill arg --- bigcodebench/provider/utility.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/bigcodebench/provider/utility.py b/bigcodebench/provider/utility.py index d363533..9c5b48f 100644 --- a/bigcodebench/provider/utility.py +++ b/bigcodebench/provider/utility.py @@ -28,8 +28,8 @@ def make_raw_chat_prompt( split: str, instruction_prefix: str, response_prefix: str, - prefill: bool, tokenizer: AutoTokenizer, + prefill: bool = True, direct_completion: bool = False, ) -> str: # directly return prompt if it does not have a tokenizer.chat_template