Update src/llama-vocab.cpp
Co-authored-by: Sigbjørn Skjæret <sigbjorn.skjaeret@scala.com>
This commit is contained in:
parent
97e1de457c
commit
4187cf5a94
|
|
@ -2505,8 +2505,8 @@ void llama_vocab::impl::load(llama_model_loader & ml, const LLM_KV & kv) {
|
||||||
for (const auto * token : {"<unk>", "<s>", "<|endoftext|>"}) {
|
for (const auto * token : {"<unk>", "<s>", "<|endoftext|>"}) {
|
||||||
_set_token_attr(token, LLAMA_TOKEN_ATTR_RSTRIP, false);
|
_set_token_attr(token, LLAMA_TOKEN_ATTR_RSTRIP, false);
|
||||||
}
|
}
|
||||||
} else if ( _contains_any(model_name, {"modern-bert"})) {
|
} else if (_contains_any(model_name, {"modern-bert"})) {
|
||||||
if ( token_to_id.count("MASK") == 0 ) {
|
if (token_to_id.count("[MASK]") == 0 ) {
|
||||||
LLAMA_LOG_WARN("%s: Mask token missing in vocab!\n", __func__);
|
LLAMA_LOG_WARN("%s: Mask token missing in vocab!\n", __func__);
|
||||||
}
|
}
|
||||||
else {
|
else {
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue