tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Wing Lian
2024-05-29 10:21:11 -04:00
committed by GitHub
parent fe650dd326
commit b7520801a3
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/axolotl/prompt_strategies/chat_template.py
View File

@@ -32,6 +32,7 @@ class ChatTemplatePrompter(Prompter):
"user": "user", "user": "user",
"assistant": "assistant", "assistant": "assistant",
"gpt": "assistant", "gpt": "assistant",
"system": "system",
} }
self.message_field_role = message_field_role self.message_field_role = message_field_role
self.message_field_content = message_field_content self.message_field_content = message_field_content