File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -6604,8 +6604,7 @@ static void llm_load_vocab(
6604
6604
vocab.type_pre = LLAMA_VOCAB_PRE_TYPE_COMMAND_R;
6605
6605
vocab.tokenizer_clean_spaces = false;
6606
6606
} else if (
6607
- tokenizer_pre == "qwen2" ||
6608
- tokenizer_pre == "megrez") {
6607
+ tokenizer_pre == "qwen2") {
6609
6608
vocab.type_pre = LLAMA_VOCAB_PRE_TYPE_QWEN2;
6610
6609
vocab.tokenizer_clean_spaces = false;
6611
6610
} else if (
@@ -6665,6 +6664,9 @@ static void llm_load_vocab(
6665
6664
} else if (
6666
6665
tokenizer_pre == "minerva-7b") {
6667
6666
vocab.type_pre = LLAMA_VOCAB_PRE_TYPE_MINERVA;
6667
+ } else if (
6668
+ tokenizer_pre == "megrez") {
6669
+ vocab.type_pre = LLAMA_VOCAB_PRE_TYPE_QWEN2;
6668
6670
} else {
6669
6671
throw std::runtime_error(format("unknown pre-tokenizer type: '%s'", tokenizer_pre.c_str()));
6670
6672
}
You can’t perform that action at this time.
0 commit comments