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;
|
int32_t res;
|
||||||
|
|
||||||
// test invalid chat template
|
// 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);
|
assert(res < 0);
|
||||||
|
|
||||||
for (size_t i = 0; i < templates.size(); i++) {
|
for (size_t i = 0; i < templates.size(); i++) {
|
||||||
|
@ -128,6 +128,8 @@ int main(void) {
|
||||||
res = llama_chat_apply_template(
|
res = llama_chat_apply_template(
|
||||||
nullptr,
|
nullptr,
|
||||||
custom_template.c_str(),
|
custom_template.c_str(),
|
||||||
|
nullptr,
|
||||||
|
nullptr,
|
||||||
conversation,
|
conversation,
|
||||||
message_count,
|
message_count,
|
||||||
true,
|
true,
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue