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
|