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

Apply suggestions from code review

Browse source
This commit is contained in:
Xuan Son Nguyen 2024-10-28 16:09:26 +01:00 • committed by GitHub
parent 839cf4ccab
commit 8b0b64bb75
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/llama.cpp
View file

@ -21716,9 +21716,9 @@ static int32_t llama_chat_apply_template_internal(
} else if (tmpl == "granite" || tmpl_contains("<|start_of_role|>")) {
// IBM Granite template
for (const auto & message: chat) {
std::string role(message -> role);
std::string role(message->role);
ss << "<|start_of_role|>" << role << "<|end_of_role|>"
<< message -> content << "<|end_of_text|>\n";
<< message->content << "<|end_of_text|>\n";
}
if (add_ass) {
ss << "<|start_of_role|>assistant<|end_of_role|>\n";