vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

test-chat-template: remove dedundant vector

Browse source
This commit is contained in:
ngxson 2024-02-16 16:33:49 +01:00
parent 4e644408a5
commit bba75c792f
1 changed files with 0 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

5
tests/test-chat-template.cpp
View file

@ -18,11 +18,6 @@ int main(void) {
{"user", "Another question"}, {"user", "Another question"},
}; };
size_t message_count = 6; size_t message_count = 6;
std::vector<const llama_chat_message *> conversation_vec;
conversation_vec.resize(message_count);
for (size_t i = 0; i < message_count; i++) {
conversation_vec[i] = &conversation[i];
}
std::vector<std::string> templates = { std::vector<std::string> templates = {
// teknium/OpenHermes-2.5-Mistral-7B // teknium/OpenHermes-2.5-Mistral-7B
"{% for message in messages %}{{'<|im_start|>' + message['role'] + '\\n' + message['content'] + '<|im_end|>' + '\\n'}}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>assistant\\n' }}{% endif %}", "{% for message in messages %}{{'<|im_start|>' + message['role'] + '\\n' + message['content'] + '<|im_end|>' + '\\n'}}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>assistant\\n' }}{% endif %}",