From b7520801a34c8b5f25ab9f0342fda11e2f9800e7 Mon Sep 17 00:00:00 2001
From: Wing Lian <wing.lian@gmail.com>
Date: Wed, 29 May 2024 10:21:11 -0400
Subject: [PATCH] handle the system role too for chat templates (#1671)

---
 src/axolotl/prompt_strategies/chat_template.py | 1 +
 1 file changed, 1 insertion(+)

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