|
|
|
@ -1,35 +1,32 @@ |
|
|
|
From d02a06f3f45a09255ace8684a66590e06ce44605 Mon Sep 17 00:00:00 2001 |
|
|
|
From: Michael Yang <mxyng@pm.me> |
|
|
|
Date: Thu, 23 May 2024 11:33:20 -0700 |
|
|
|
Subject: [PATCH] default pretokenizer on unrecognized type |
|
|
|
|
|
|
|
---
|
|
|
|
llama.cpp | 5 +---- |
|
|
|
1 file changed, 1 insertion(+), 4 deletions(-) |
|
|
|
|
|
|
|
diff --git a/llama.cpp b/llama.cpp
|
|
|
|
index 15c66077..af1aede3 100644
|
|
|
|
index 40d2ec2c..74f3ee9c 100644
|
|
|
|
--- a/llama.cpp
|
|
|
|
+++ b/llama.cpp
|
|
|
|
@@ -4504,9 +4504,6 @@ static void llm_load_vocab(
|
|
|
|
LLAMA_LOG_WARN("%s: ************************************ \n", __func__); |
|
|
|
LLAMA_LOG_WARN("%s: \n", __func__); |
|
|
|
vocab.type_pre = LLAMA_VOCAB_PRE_TYPE_DEFAULT; |
|
|
|
- } else if (
|
|
|
|
- tokenizer_pre == "default") {
|
|
|
|
@@ -4642,16 +4642,7 @@ static void llm_load_vocab(
|
|
|
|
|
|
|
|
// for now, only BPE models have pre-tokenizers |
|
|
|
if (vocab.type == LLAMA_VOCAB_TYPE_BPE) { |
|
|
|
- if (tokenizer_pre.empty()) {
|
|
|
|
- LLAMA_LOG_WARN("%s: missing pre-tokenizer type, using: 'default'\n", __func__);
|
|
|
|
- LLAMA_LOG_WARN("%s: \n", __func__);
|
|
|
|
- LLAMA_LOG_WARN("%s: ************************************ \n", __func__);
|
|
|
|
- LLAMA_LOG_WARN("%s: GENERATION QUALITY WILL BE DEGRADED! \n", __func__);
|
|
|
|
- LLAMA_LOG_WARN("%s: CONSIDER REGENERATING THE MODEL \n", __func__);
|
|
|
|
- LLAMA_LOG_WARN("%s: ************************************ \n", __func__);
|
|
|
|
- LLAMA_LOG_WARN("%s: \n", __func__);
|
|
|
|
- vocab.type_pre = LLAMA_VOCAB_PRE_TYPE_DEFAULT;
|
|
|
|
- } else if (
|
|
|
|
+ if (
|
|
|
|
tokenizer_pre == "default") { |
|
|
|
vocab.type_pre = LLAMA_VOCAB_PRE_TYPE_DEFAULT; |
|
|
|
} else if ( |
|
|
|
tokenizer_pre == "llama3" || |
|
|
|
tokenizer_pre == "llama-v3" || |
|
|
|
@@ -4553,7 +4550,7 @@ static void llm_load_vocab(
|
|
|
|
tokenizer_pre == "dbrx") { |
|
|
|
vocab.type_pre = LLAMA_VOCAB_PRE_TYPE_DBRX; |
|
|
|
@@ -4703,7 +4694,8 @@ static void llm_load_vocab(
|
|
|
|
tokenizer_pre == "smaug-bpe") { |
|
|
|
vocab.type_pre = LLAMA_VOCAB_PRE_TYPE_SMAUG; |
|
|
|
} else { |
|
|
|
- throw std::runtime_error(format("unknown pre-tokenizer type: '%s'", tokenizer_pre.c_str()));
|
|
|
|
+ LLAMA_LOG_WARN("%s: missing or unrecognized pre-tokenizer type, using: 'default'\n", __func__);
|
|
|
|
+ vocab.type_pre = LLAMA_VOCAB_PRE_TYPE_DEFAULT;
|
|
|
|
} |
|
|
|
} else { |
|
|
|
vocab.type_pre = LLAMA_VOCAB_PRE_TYPE_DEFAULT; |
|
|
|
--
|
|
|
|
2.45.1 |
|
|
|
|
|
|
|
|