tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 3 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

9
src/axolotl/prompters.py
View File

@@ -352,9 +352,12 @@ 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."
) )
role = CONVERSATION_ROLE_FORMAT[self._conversation.name].format( if self._conversation.name in ["llama3"]:
ROLE=from_role role = from_role
) else:
role = CONVERSATION_ROLE_FORMAT[self._conversation.name].format(
ROLE=from_role
)
if len(conv.messages) > 0 and ((role == conv.messages[-1][0])): if len(conv.messages) > 0 and ((role == conv.messages[-1][0])):
if ( if (