Sfoglia il codice sorgente

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

0xspringtime 1 anno fa
parent
commit
3aa184a8c7
1 ha cambiato i file con 7 aggiunte e 1 eliminazioni
  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()