From 1b7e8604bb6eec2ec5da93fb1261b98a1176ff11 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Wed, 16 Aug 2023 11:21:03 -0400 Subject: [PATCH] fix orca prompts (#422) --- src/axolotl/prompt_strategies/alpaca_w_system.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/axolotl/prompt_strategies/alpaca_w_system.py b/src/axolotl/prompt_strategies/alpaca_w_system.py index d56520bd7..4b5521d48 100644 --- a/src/axolotl/prompt_strategies/alpaca_w_system.py +++ b/src/axolotl/prompt_strategies/alpaca_w_system.py @@ -92,8 +92,9 @@ class OpenOrcaSystemDataPrompter(SystemDataPrompter): def match_prompt_style(self): # pylint: disable=duplicate-code if self.prompt_style == PromptStyle.INSTRUCT.value: - self.turn_format = "### User:\n{instruction}\n\n### Additional Context:\n{input}\n\n### Assistant:\n" - self.turn_no_input_format = "### User:\n{instruction}\n\n### Assistant:\n" + self.turn_format = "### Human:\n{instruction}\n### Additional Context:\n{input}\n### Assistant:\n" + self.turn_no_input_format = "### Human:\n{instruction}\n### Assistant:\n" + self.system_format = "### System:\n{system}\n" if self.prompt_style == PromptStyle.CHAT.value: self.turn_format = "USER: {instruction}\n{input}\nASSISTANT:" self.turn_no_input_format = "USER: {instruction}\nASSISTANT:"