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
|