fix prefill arg

bigcode-project · Feb 1, 2025 · dcff46f · dcff46f
1 parent 1a34078
commit dcff46f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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