mirror of
https://github.com/ggml-org/llama.cpp.git
synced 2025-11-04 09:32:00 +00:00
Merge branch 'add-fh1-rebased' of https://github.com/tiiuae/llama.cpp-public into add-fh1-rebased
This commit is contained in:
@@ -607,7 +607,14 @@ class TextModel(ModelBase):
|
||||
|
||||
from transformers import AutoTokenizer
|
||||
tokenizer = AutoTokenizer.from_pretrained(self.dir_model)
|
||||
<<<<<<< HEAD
|
||||
vocab_size = self.hparams.get("vocab_size", len(tokenizer.vocab))
|
||||
=======
|
||||
vocab_size = min(
|
||||
self.hparams.get("vocab_size", len(tokenizer.vocab)),
|
||||
len(tokenizer.vocab)
|
||||
)
|
||||
>>>>>>> 250b4f1074da288399e71577a925aa83acf316d5
|
||||
assert max(tokenizer.vocab.values()) < vocab_size
|
||||
|
||||
tokpre = self.get_vocab_base_pre(tokenizer)
|
||||
|
||||
Reference in New Issue
Block a user