Skip to content

Commit 098ee79

Browse files
committed
minor fix
1 parent 8993dca commit 098ee79

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

bigcodebench/provider/utility.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -60,9 +60,8 @@ def make_raw_chat_prompt(
6060
task_prompt = tokenizer.apply_chat_template(
6161
[
6262
{"role": "user", "content": task_prompt},
63-
# {"role": "assistant", "content": response},
63+
{"role": "assistant", "content": response},
6464
],
6565
tokenize=False,
66-
add_generation_prompt=True
6766
).split(_MAGIC_SPLITTER_)[0]
6867
return task_prompt

0 commit comments

Comments
 (0)