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

fix: test

Browse Source
This commit is contained in:
NanoCode012
2024-10-11 11:11:19 +07:00
parent 2038255052
commit e3efa29cf5
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/prompt_strategies/test_chat_templates.py
View File

@@ -73,7 +73,7 @@ class TestAssistantChatTemplateLlama3:
strategy = ChatTemplateStrategy( strategy = ChatTemplateStrategy(
ChatTemplatePrompter( ChatTemplatePrompter(
llama3_tokenizer, llama3_tokenizer,
chat_templates("llama3"), chat_template=get_chat_template("llama3"),
message_field_role="role", message_field_role="role",
message_field_content="content", message_field_content="content",
roles={ roles={