winglian commited on
Commit
1b7e860
1 Parent(s): 3d1f203

fix orca prompts (#422)

Browse files
src/axolotl/prompt_strategies/alpaca_w_system.py CHANGED
@@ -92,8 +92,9 @@ class OpenOrcaSystemDataPrompter(SystemDataPrompter):
92
  def match_prompt_style(self):
93
  # pylint: disable=duplicate-code
94
  if self.prompt_style == PromptStyle.INSTRUCT.value:
95
- self.turn_format = "### User:\n{instruction}\n\n### Additional Context:\n{input}\n\n### Assistant:\n"
96
- self.turn_no_input_format = "### User:\n{instruction}\n\n### Assistant:\n"
 
97
  if self.prompt_style == PromptStyle.CHAT.value:
98
  self.turn_format = "USER: {instruction}\n{input}\nASSISTANT:"
99
  self.turn_no_input_format = "USER: {instruction}\nASSISTANT:"
 
92
  def match_prompt_style(self):
93
  # pylint: disable=duplicate-code
94
  if self.prompt_style == PromptStyle.INSTRUCT.value:
95
+ self.turn_format = "### Human:\n{instruction}\n### Additional Context:\n{input}\n### Assistant:\n"
96
+ self.turn_no_input_format = "### Human:\n{instruction}\n### Assistant:\n"
97
+ self.system_format = "### System:\n{system}\n"
98
  if self.prompt_style == PromptStyle.CHAT.value:
99
  self.turn_format = "USER: {instruction}\n{input}\nASSISTANT:"
100
  self.turn_no_input_format = "USER: {instruction}\nASSISTANT:"