05-default-pretokenizer.diff 1.9 KB

123456789101112131415161718192021222324252627282930313233
  1. diff --git a/src/llama.cpp b/src/llama.cpp
  2. index 2b9ace28..172640e2 100644
  3. --- a/src/llama.cpp
  4. +++ b/src/llama.cpp
  5. @@ -5357,16 +5357,7 @@ static void llm_load_vocab(
  6. if (vocab.type == LLAMA_VOCAB_TYPE_BPE) {
  7. vocab.tokenizer_add_space_prefix = false;
  8. vocab.tokenizer_clean_spaces = true;
  9. - if (tokenizer_pre.empty()) {
  10. - LLAMA_LOG_WARN("%s: missing pre-tokenizer type, using: 'default'\n", __func__);
  11. - LLAMA_LOG_WARN("%s: \n", __func__);
  12. - LLAMA_LOG_WARN("%s: ************************************ \n", __func__);
  13. - LLAMA_LOG_WARN("%s: GENERATION QUALITY WILL BE DEGRADED! \n", __func__);
  14. - LLAMA_LOG_WARN("%s: CONSIDER REGENERATING THE MODEL \n", __func__);
  15. - LLAMA_LOG_WARN("%s: ************************************ \n", __func__);
  16. - LLAMA_LOG_WARN("%s: \n", __func__);
  17. - vocab.type_pre = LLAMA_VOCAB_PRE_TYPE_DEFAULT;
  18. - } else if (tokenizer_pre == "default") {
  19. + if (tokenizer_pre == "default") {
  20. vocab.type_pre = LLAMA_VOCAB_PRE_TYPE_DEFAULT;
  21. } else if (
  22. tokenizer_pre == "llama3" ||
  23. @@ -5439,7 +5430,8 @@ static void llm_load_vocab(
  24. tokenizer_pre == "jais") {
  25. vocab.type_pre = LLAMA_VOCAB_PRE_TYPE_JAIS;
  26. } else {
  27. - throw std::runtime_error(format("unknown pre-tokenizer type: '%s'", tokenizer_pre.c_str()));
  28. + LLAMA_LOG_WARN("%s: missing or unrecognized pre-tokenizer type, using: 'default'\n", __func__);
  29. + vocab.type_pre = LLAMA_VOCAB_PRE_TYPE_DEFAULT;
  30. }
  31. } else if (vocab.type == LLAMA_VOCAB_TYPE_SPM) {
  32. vocab.type_pre = LLAMA_VOCAB_PRE_TYPE_DEFAULT;