Skip to content

Commit 4fc3776

Browse files
committed
Add another test case
1 parent 6f7daba commit 4fc3776

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

tests/test-tokenizer-0.cpp

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,8 +14,9 @@ static const std::map<std::string, std::vector<llama_token>> & k_tests()
1414
{ " this is 🦙.cpp", { 1, 445, 338, 29871, 243, 162, 169, 156, 29889, 8223, }, },
1515
{ "w048 7tuijk dsdfhu", { 1, 29893, 29900, 29946, 29947, 29871, 29955, 9161, 13535, 18031, 2176, 6905, }, },
1616
{ "нещо на Български", { 1, 821, 4851, 665, 1386, 29713, 1305, }, },
17-
{ "<🦙>test extra_id_1 test", { 1, 32003, 1688, 29871, 32001, 259, 1243, }, },
18-
{ "<🦙>test extra_id_100 test", { 1, 32003, 1688, 29871, 32002, 1243, }, },
17+
{ "<🦙>test extra_id_1 test", { 1, 32004, 1688, 29871, 32001, 259, 1243, }, },
18+
{ "<🦙>test extra_id_100 test", { 1, 32004, 1688, 29871, 32002, 1243, }, },
19+
{ "<🦙>test extra_id_200 test", { 1, 32004, 1688, 321, 32003, 1243, }, },
1920
};
2021
return _k_tests;
2122
};
@@ -50,7 +51,8 @@ int main(int argc, char **argv) {
5051

5152
llama_add_special_token(model, "extra_id_1", 32001);
5253
llama_add_special_token(model, "extra_id_100", 32002);
53-
llama_add_special_token(model, "<🦙>", 32003);
54+
llama_add_special_token(model, "xtra_id_200", 32003);
55+
llama_add_special_token(model, "<🦙>", 32004);
5456

5557
ctx = llama_new_context_with_model(model, lparams);
5658

0 commit comments

Comments
 (0)