Просмотр исходного кода

convert-hf : change assert to exception (#8015)

0xspringtime 1 год назад
Родитель
Сommit
3aa184a8c7
1 измененных файлов с 7 добавлено и 1 удалено
  1. 7 1
      convert-hf-to-gguf.py

+ 7 - 1
convert-hf-to-gguf.py

@@ -967,7 +967,13 @@ class XverseModel(Model):
         from transformers import AutoTokenizer
         tokenizer = AutoTokenizer.from_pretrained(dir_model)
         vocab_size = hparams.get("vocab_size", len(tokenizer.vocab))
-        assert max(tokenizer.vocab.values()) < vocab_size
+        # Since we are checking the maximum index, we need to ensure it's strictly less than vocab_size,
+        # because vocab_size is the count of items, and indexes start at 0.
+        max_vocab_index = max(tokenizer.get_vocab().values())
+        if max_vocab_index >= vocab_size:
+            raise ValueError("Vocabulary size exceeds expected maximum size.")
+
+
 
         reverse_vocab: dict[int, str] = {id_: encoded_tok for encoded_tok, id_ in tokenizer.vocab.items()}
         added_vocab = tokenizer.get_added_vocab()