0005-default-pretokenizer.patch 2.2 KB
Newer Older
1
From 0e531d69786c4a96a3a2bcf7b2d576bd6f7edf25 Mon Sep 17 00:00:00 2001
Michael Yang's avatar
Michael Yang committed
2
3
4
5
6
7
8
9
From: Michael Yang <mxyng@pm.me>
Date: Mon, 16 Sep 2024 15:53:13 -0700
Subject: [PATCH] 05-default-pretokenizer.diff

---
 src/llama.cpp | 14 +++-----------
 1 file changed, 3 insertions(+), 11 deletions(-)

10
diff --git a/src/llama.cpp b/src/llama.cpp
11
index 4c0a1bb6..800dfb95 100644
12
13
--- a/src/llama.cpp
+++ b/src/llama.cpp
14
@@ -6287,16 +6287,7 @@ static void llm_load_vocab(
15
         if (vocab.type == LLAMA_VOCAB_TYPE_BPE) {
16
17
             vocab.tokenizer_add_space_prefix = false;
             vocab.tokenizer_clean_spaces = true;
18
19
20
21
22
23
24
25
-            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__);
Michael Yang's avatar
Michael Yang committed
26
-                vocab.type_pre = LLAMA_VOCAB_PRE_TYPE_DEFAULT;
27
28
-            } else if (tokenizer_pre == "default") {
+            if (tokenizer_pre == "default") {
29
                 vocab.type_pre = LLAMA_VOCAB_PRE_TYPE_DEFAULT;
Michael Yang's avatar
Michael Yang committed
30
             } else if (
31
                     tokenizer_pre == "llama3"   ||
32
33
34
@@ -6398,7 +6389,8 @@ static void llm_load_vocab(
                 vocab.tokenizer_add_bos = true;
                 vocab.tokenizer_clean_spaces = false;
Michael Yang's avatar
Michael Yang committed
35
36
             } else {
-                throw std::runtime_error(format("unknown pre-tokenizer type: '%s'", tokenizer_pre.c_str()));
37
+                LLAMA_LOG_WARN("%s: missing or unrecognized pre-tokenizer type, using: 'default'\n", __func__);
Michael Yang's avatar
Michael Yang committed
38
39
+                vocab.type_pre = LLAMA_VOCAB_PRE_TYPE_DEFAULT;
             }
40
         } else if (vocab.type == LLAMA_VOCAB_TYPE_SPM) {
Michael Yang's avatar
Michael Yang committed
41
             vocab.type_pre = LLAMA_VOCAB_PRE_TYPE_DEFAULT;
Michael Yang's avatar
Michael Yang committed
42
-- 
43
2.39.3 (Apple Git-146)
Michael Yang's avatar
Michael Yang committed
44