From 479c1520b1d7edce84625e755012f9811c24266c Mon Sep 17 00:00:00 2001 From: ochafik Date: Thu, 31 Oct 2024 14:49:59 +0000 Subject: [PATCH] `tool-call`: fix qwen template test --- tests/test-tool-call.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test-tool-call.cpp b/tests/test-tool-call.cpp index a39b1d65f..c81a4c15a 100644 --- a/tests/test-tool-call.cpp +++ b/tests/test-tool-call.cpp @@ -407,7 +407,7 @@ static void test_grammars() { test_template("tests/chat/templates/mistralai-Mistral-Nemo-Instruct-2407.jinja", "", "", { "" }, tool_call_message_with_id, tools, /* skip_grammar_test= */ true); - test_template("tests/chat/templates/Qwen-Qwen2.5-7B-Instruct.jinja", "", "", { "" }, tool_call_message, tools); + test_template("tests/chat/templates/Qwen-Qwen2.5-7B-Instruct.jinja", "", "", { "<|im_end|>" }, tool_call_message, tools); test_template("tests/chat/templates/NousResearch-Hermes-2-Pro-Llama-3-8B-tool_use.jinja", "", "", { "<|im_end|>" }, tool_call_message, tools); test_template("tests/chat/templates/NousResearch-Hermes-3-Llama-3.1-8B-tool_use.jinja", "", "", { "<|im_end|>" }, tool_call_message, tools); test_template("tests/chat/templates/meta-llama-Meta-Llama-3.1-8B-Instruct.jinja", "", "", { "<|eom_id|>", "<|eot_id|>" }, tool_call_message, tools);