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

Update convert_hf_to_gguf.py

Browse source 
Co-authored-by: compilade <git@compilade.net>
This commit is contained in:
Yoshi Suhara 2024-08-09 02:05:50 -07:00 • committed by Yoshi Suhara
parent 147cdf641a
commit b841554d0c
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

@ -3740,7 +3740,7 @@ class NemotronModel(Model):
model_arch = gguf.MODEL_ARCH.NEMOTRON model_arch = gguf.MODEL_ARCH.NEMOTRON
def set_vocab(self): def set_vocab(self):
self. _set_vocab_sentencepiece() self._set_vocab_sentencepiece()
self.gguf_writer.add_pad_token_id(0) self.gguf_writer.add_pad_token_id(0)
self.gguf_writer.add_unk_token_id(1) self.gguf_writer.add_unk_token_id(1)