summaryrefslogtreecommitdiff
path: root/llama.h
diff options
context:
space:
mode:
authorJared Van Bortel <jared@nomic.ai>2024-03-28 11:44:36 -0400
committerGitHub <noreply@github.com>2024-03-28 11:44:36 -0400
commitbe55134a535f7218c53f39211755b1c7550851b2 (patch)
treef27a84973005f1e46d07323bdd1f983a11327b05 /llama.h
parent66ba56025602270152f5ba5234f3a80be3dee1c9 (diff)
convert : refactor vocab selection logic (#6355)
Diffstat (limited to 'llama.h')
-rw-r--r--llama.h6
1 files changed, 3 insertions, 3 deletions
diff --git a/llama.h b/llama.h
index 1fe4af49..f061d014 100644
--- a/llama.h
+++ b/llama.h
@@ -60,9 +60,9 @@ extern "C" {
enum llama_vocab_type {
LLAMA_VOCAB_TYPE_NONE = 0, // For models without vocab
- LLAMA_VOCAB_TYPE_SPM = 1, // SentencePiece
- LLAMA_VOCAB_TYPE_BPE = 2, // Byte Pair Encoding
- LLAMA_VOCAB_TYPE_WPM = 3, // WordPiece
+ LLAMA_VOCAB_TYPE_SPM = 1, // LLaMA tokenizer based on byte-level BPE with byte fallback
+ LLAMA_VOCAB_TYPE_BPE = 2, // GPT-2 tokenizer based on byte-level BPE
+ LLAMA_VOCAB_TYPE_WPM = 3, // BERT tokenizer based on WordPiece
};
// note: these values should be synchronized with ggml_rope