From fc050381ef62102f277bb20657a53c691281b047 Mon Sep 17 00:00:00 2001 From: MaggotHATE Date: Wed, 20 Nov 2024 16:12:25 +0500 Subject: [PATCH] Fixed `test-chat-template` --- tests/test-chat-template.cpp | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/test-chat-template.cpp b/tests/test-chat-template.cpp index 03e897e66..51121533c 100644 --- a/tests/test-chat-template.cpp +++ b/tests/test-chat-template.cpp @@ -118,7 +118,7 @@ int main(void) { int32_t res; // test invalid chat template - res = llama_chat_apply_template(nullptr, "INVALID TEMPLATE", conversation, message_count, true, formatted_chat.data(), formatted_chat.size()); + res = llama_chat_apply_template(nullptr, "INVALID TEMPLATE", nullptr, nullptr, conversation, message_count, true, formatted_chat.data(), formatted_chat.size()); assert(res < 0); for (size_t i = 0; i < templates.size(); i++) { @@ -128,6 +128,8 @@ int main(void) { res = llama_chat_apply_template( nullptr, custom_template.c_str(), + nullptr, + nullptr, conversation, message_count, true,