Fixed test-chat-template
This commit is contained in:
parent
b3e343eae7
commit
fc050381ef
1 changed files with 3 additions and 1 deletions
|
@ -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,
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue