Merge pull request #75 from zhangchen-xu/main

terryyz · web-flow · commit 4381f0afa384 · 2025-02-08T18:40:24.000+08:00
fix make_raw_chat_prompt when prefill is disabled
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`