handle the system role too for chat templates (#1671)

2024-05-29 10:21:11 -04:00
parent fe650dd326
commit b7520801a3
1 changed files with 1 additions and 0 deletions
--- 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