root/llama.cpp
1
0
Fork 0
mirror of https://github.com/ggerganov/llama.cpp.git synced 2024-12-30 21:34:36 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

style

Browse Source
This commit is contained in:
Pedro Cuenca 2024-04-18 18:46:03 +02:00
parent d79ab101c3
commit 112c4c4e9b
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
convert-hf-to-gguf.py
View File

@ -1309,8 +1309,10 @@ class LlamaModel(Model):
# Apply to CodeLlama only (and ignore for Llama 3 with a vocab size of 128256) # Apply to CodeLlama only (and ignore for Llama 3 with a vocab size of 128256)
if self.hparams.get("vocab_size", 32000) == 32016: if self.hparams.get("vocab_size", 32000) == 32016:
special_vocab = gguf.SpecialVocab(self.dir_model, load_merges=False, special_vocab = gguf.SpecialVocab(
special_token_types = ['prefix', 'suffix', 'middle', 'eot']) self.dir_model, load_merges=False,
special_token_types = ['prefix', 'suffix', 'middle', 'eot']
)
special_vocab._set_special_token("prefix", 32007) special_vocab._set_special_token("prefix", 32007)
special_vocab._set_special_token("suffix", 32008) special_vocab._set_special_token("suffix", 32008)
special_vocab._set_special_token("middle", 32009) special_vocab._set_special_token("middle", 32009)