Merge branch 'main' into cj_tokenizer_default_prompt_template

2024-10-10 20:25:37 +07:00
parent dab2590e4d 6d3caadf90
commit 2038255052
21 changed files with 357 additions and 20 deletions
--- a/tests/prompt_strategies/test_chat_templates.py
+++ b/tests/prompt_strategies/test_chat_templates.py
@@ -73,7 +73,7 @@ class TestAssistantChatTemplateLlama3:
        strategy = ChatTemplateStrategy(
            ChatTemplatePrompter(
                llama3_tokenizer,
-                chat_template=get_chat_template("llama3"),
+                chat_templates("llama3"),
                message_field_role="role",
                message_field_content="content",
                roles={