Use sentencepiece tokenizer, or fall back to hfft.
This commit is contained in:
parent
23ffda00df
commit
16ede02a47
1 changed files with 4 additions and 1 deletions
|
@ -1057,7 +1057,10 @@ class LlamaModel(Model):
|
|||
model_arch = gguf.MODEL_ARCH.LLAMA
|
||||
|
||||
def set_vocab(self):
|
||||
self._set_vocab_llama_hf()
|
||||
try:
|
||||
self. _set_vocab_sentencepiece()
|
||||
except FileNotFoundError:
|
||||
self._set_vocab_llama_hf()
|
||||
|
||||
def set_gguf_parameters(self):
|
||||
super().set_gguf_parameters()
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue