Skip to content

Commit 0261c52

Browse files
authored
add support for codellama-70b prompt
1 parent 00967d1 commit 0261c52

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

bigcode_eval/tasks/humanevalpack.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -225,6 +225,8 @@ def get_prompt(self, prompt_base, instruction, context=None):
225225
prompt = f'Below is an instruction that describes a task. Write a response that appropriately completes the request.\n\n### Instruction:\n{inp}\n\n### Response:\n{prompt_base}'
226226
elif self.prompt == "codellama":
227227
prompt = f"[INST] {inp.strip()} [/INST] {prompt_base}"
228+
elif self.prompt == "codellama-70b":
229+
prompt = f"Source: user\n\n {inp.strip()} Source: assistant\nDestination: user \n\n{prompt_base}"
228230
else:
229231
raise ValueError(f"The --prompt argument {self.prompt} wasn't provided or isn't supported")
230232
# Strip off the final \n to make the tokens more natural

0 commit comments

Comments
 (0)