diff --git a/src/axolotl/prompt_strategies/chat_template.py b/src/axolotl/prompt_strategies/chat_template.py index b052469dc..0e7d823ed 100644 --- a/src/axolotl/prompt_strategies/chat_template.py +++ b/src/axolotl/prompt_strategies/chat_template.py @@ -32,6 +32,7 @@ class ChatTemplatePrompter(Prompter): "user": "user", "assistant": "assistant", "gpt": "assistant", + "system": "system", } self.message_field_role = message_field_role self.message_field_content = message_field_content