Apply suggestion

Co-authored-by: Xuan Son Nguyen <thichthat@gmail.com>
This commit is contained in:
Tristan Druyen 2024-05-23 14:20:48 +02:00 committed by GitHub
parent 037af53bfa
commit 05e6bc6c55
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -17628,7 +17628,7 @@ static int32_t llama_chat_apply_template_internal(
} }
} }
// llama2 templates seem to not care about "add_generation_prompt" // llama2 templates seem to not care about "add_generation_prompt"
} else if (tmpl == "phi3" || (tmpl.find("<|assistant|>") != std::string::npos && tmpl.find("<|end|>") != std::string::npos )) { } else if (tmpl == "phi3" || (tmpl.find("<|assistant|>") != std::string::npos && tmpl.find("<|end|>") != std::string::npos)) {
// Phi 3 // Phi 3
for (auto message : chat) { for (auto message : chat) {
std::string role(message->role); std::string role(message->role);