vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

model: dbrx: convert-hf-to-gguf.py support python 3.8

Browse source
This commit is contained in:
Pierrick HYMBERT 2024-04-07 17:25:39 +02:00
parent 3a9dc2eee2
commit 8154617ff2
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
convert-hf-to-gguf.py
View file

@ -416,7 +416,7 @@ class Model(ABC):
# for this kind of tokenizer, added_vocab is not a subset of vocab, so they need to be combined
added_vocab = tokenizer.get_added_vocab()
reverse_vocab = {id_: encoded_tok for encoded_tok, id_ in (vocab | added_vocab).items()}
reverse_vocab = {id_: encoded_tok for encoded_tok, id_ in ({**vocab, **added_vocab}).items()}
for i in range(vocab_size):
if i not in reverse_vocab: