Skip to content

Commit 459df19

Browse files
LLM_CHAT_TEMPLATE_KIMI_K2 if(add_ass){}
1 parent 0a5704b commit 459df19

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/llama-chat.cpp

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -698,6 +698,10 @@ int32_t llm_chat_apply_template(
698698
}
699699

700700
ss << message->content << "<|im_end|>";
701+
702+
if (add_ass) {
703+
ss << "<|im_assistant|>assistant<|im_middle|>";
704+
}
701705
}
702706
} else {
703707
// template not supported

0 commit comments

Comments
 (0)