llama-chat : fix multiple system message for gemma, orion (#14246)
This commit is contained in:
parent
3865cff4f5
commit
95402553a5
1 changed files with 2 additions and 2 deletions
|
@ -333,7 +333,7 @@ int32_t llm_chat_apply_template(
|
||||||
std::string role(message->role);
|
std::string role(message->role);
|
||||||
if (role == "system") {
|
if (role == "system") {
|
||||||
// there is no system message for gemma, but we will merge it with user prompt, so nothing is broken
|
// there is no system message for gemma, but we will merge it with user prompt, so nothing is broken
|
||||||
system_prompt = trim(message->content);
|
system_prompt += trim(message->content);
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
// in gemma, "assistant" is "model"
|
// in gemma, "assistant" is "model"
|
||||||
|
@ -355,7 +355,7 @@ int32_t llm_chat_apply_template(
|
||||||
std::string role(message->role);
|
std::string role(message->role);
|
||||||
if (role == "system") {
|
if (role == "system") {
|
||||||
// there is no system message support, we will merge it with user prompt
|
// there is no system message support, we will merge it with user prompt
|
||||||
system_prompt = message->content;
|
system_prompt += message->content;
|
||||||
continue;
|
continue;
|
||||||
} else if (role == "user") {
|
} else if (role == "user") {
|
||||||
ss << "Human: ";
|
ss << "Human: ";
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue