Merge pull request #208 from bigcode-project/aurora-prompt

Add prompt
bigcode-project · Mar 27, 2024 · 094c7cc · 094c7cc
2 parents e54f33d + 974c7ed
commit 094c7cc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/bigcode_eval/tasks/humanevalpack.py b/bigcode_eval/tasks/humanevalpack.py
@@ -239,6 +239,8 @@ def get_prompt(self, prompt_base, instruction, context=None):
             prompt = f"<|im_start|>user\n{inp}<|im_end|>\n<|im_start|>assistant\n{prompt_base}"
         elif self.prompt == "codellama-70b":
             prompt = f"Source: user\n\n {inp.strip()} Source: assistant\nDestination: user \n\n{prompt_base}"
+        elif self.prompt == "aurora-m":
+            prompt = f'### Instruction:\n{inp}\n### Response:\n{prompt_base}'
         else:
             raise ValueError(f"The --prompt argument {self.prompt} wasn't provided or isn't supported")
         # Strip off the final \n to make the tokens more natural