Skip to content

Commit beb6d0c

Browse files
committed
cont : add <eos> and warning
1 parent 7a497a1 commit beb6d0c

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

src/llama-vocab.cpp

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2558,7 +2558,8 @@ void llama_vocab::impl::load(llama_model_loader & ml, const LLM_KV & kv) {
25582558
|| t.first == "[EOS]" // Kimi-K2
25592559
|| t.first == "<|end_of_text|>"
25602560
|| t.first == "<end_of_utterance>" // smoldocling
2561-
|| t.first == "<turn|>" // gemma4
2561+
|| t.first == "<eos>" // gemma4
2562+
|| t.first == "<turn|>" // gemma4
25622563
|| t.first == "<|tool_response>" // gemma4
25632564
|| t.first == "<|end▁of▁sentence|>" // deepseek-ocr
25642565
) {
@@ -2668,6 +2669,8 @@ void llama_vocab::impl::load(llama_model_loader & ml, const LLM_KV & kv) {
26682669

26692670
auto & attr = id_to_token[s_id].attr;
26702671
attr = LLAMA_TOKEN_ATTR_NORMAL;
2672+
2673+
LLAMA_LOG_WARN("%s: special_eog_ids contains '<|tool_response>', removing '</s>' token from EOG list\n", __func__);
26712674
}
26722675
}
26732676
}

0 commit comments

Comments
 (0)