fix make_raw_chat_prompt when prefill is disabled

zhangchen-xu · zhangchen-xu · commit e94072459359 · 2025-02-08T00:29:37.000-08:00
diff --git a/bigcodebench/provider/utility.py b/bigcodebench/provider/utility.py
@@ -72,7 +72,7 @@ def make_raw_chat_prompt(
                 [
                     {"role": "user", "content": task_prompt},
                 ],
-                tokenize=False,
+                tokenize=False, add_generation_prompt=True
             ).split(_MAGIC_SPLITTER_)[0]
     return task_prompt
 

Original file line number	Diff line number	Diff line change
`@@ -72,7 +72,7 @@ def make_raw_chat_prompt(`
`72`	`72`	`[`
`73`	`73`	`{"role": "user", "content": task_prompt},`
`74`	`74`	`],`
`75`		`- tokenize=False,`
	`75`	`+ tokenize=False, add_generation_prompt=True`
`76`	`76`	`).split(_MAGIC_SPLITTER_)[0]`
`77`	`77`	`return task_prompt`
`78`	`78`