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:"
|