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 = "###
|
96 |
-
self.turn_no_input_format = "###
|
|
|
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:"
|