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

updating velvet chat template

Browse source
This commit is contained in:
f.buciuni 2025-02-07 19:54:55 +01:00
parent 66e6d10b61
commit 39795570db
1 changed files with 6 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/llama-chat.cpp
View file

@ -573,8 +573,11 @@ int32_t llm_chat_apply_template(
// Velvet template // Velvet template
std::string leading_space = ""; std::string leading_space = "";
std::string trailing_space = ""; std::string trailing_space = "";
bool trim_assistant_message = true; bool trim_assistant_message = false;
bool is_inside_turn = false; bool is_inside_turn = false;
std::string system_message = "";
std::string last_message(chat.back()->content);
ss << "<s>";
for (auto message : chat) { for (auto message : chat) {
if (!is_inside_turn) { if (!is_inside_turn) {
ss << leading_space << "<instruction>" << trailing_space; ss << leading_space << "<instruction>" << trailing_space;
@ -583,9 +586,9 @@ int32_t llm_chat_apply_template(
std::string role(message->role); std::string role(message->role);
std::string content(message->content); std::string content(message->content);
if (role == "system") { if (role == "system") {
ss << content << "\n\n"; system_message = content + "\n\n";
} else if (role == "user") { } else if (role == "user") {
ss << content << leading_space << "</instruction>"; ss << (content==last_message ? system_message : "") << content << leading_space << "</instruction>";
} else { } else {
ss << trailing_space << (trim_assistant_message ? trim(content) : content) << "</s>"; ss << trailing_space << (trim_assistant_message ? trim(content) : content) << "</s>";
is_inside_turn = false; is_inside_turn = false;