winglian commited on
Commit
b752080
·
unverified ·
1 Parent(s): fe650dd

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

Browse files
src/axolotl/prompt_strategies/chat_template.py CHANGED
@@ -32,6 +32,7 @@ class ChatTemplatePrompter(Prompter):
32
  "user": "user",
33
  "assistant": "assistant",
34
  "gpt": "assistant",
 
35
  }
36
  self.message_field_role = message_field_role
37
  self.message_field_content = message_field_content
 
32
  "user": "user",
33
  "assistant": "assistant",
34
  "gpt": "assistant",
35
+ "system": "system",
36
  }
37
  self.message_field_role = message_field_role
38
  self.message_field_content = message_field_content