Skip to content

Commit 2d56a29

Browse files
committed
fix regression
1 parent 5cfc73b commit 2d56a29

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

convert_hf_to_gguf.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -607,6 +607,7 @@ def get_vocab_base(self) -> tuple[list[str], list[int], str]:
607607

608608
from transformers import AutoTokenizer
609609
tokenizer = AutoTokenizer.from_pretrained(self.dir_model)
610+
vocab_size = self.hparams.get("vocab_size", len(tokenizer.vocab))
610611
assert max(tokenizer.vocab.values()) < vocab_size
611612

612613
tokpre = self.get_vocab_base_pre(tokenizer)

0 commit comments

Comments
 (0)