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

better handling of llama-3 tool rolw (#1782)

Browse Source
This commit is contained in:
Wing Lian
2024-08-25 12:31:40 -04:00
committed by GitHub
parent 22f4eafa55
commit f245964f22
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/axolotl/prompters.py
View File

@@ -352,6 +352,9 @@ class ShareGPTPrompter(Prompter): # pylint: disable=too-few-public-methods
"Please help us by creating an Issue to add support for this conversation type." "Please help us by creating an Issue to add support for this conversation type."
) )
if self._conversation.name in ["llama3"]:
role = from_role
else:
role = CONVERSATION_ROLE_FORMAT[self._conversation.name].format( role = CONVERSATION_ROLE_FORMAT[self._conversation.name].format(
ROLE=from_role ROLE=from_role
) )