Skip to content

Commit 08e0a4e

Browse files
committed
handle vocab
1 parent 1a308da commit 08e0a4e

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

convert_hf_to_gguf.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9982,6 +9982,9 @@ def __init__(self, *args, **kwargs):
99829982
config["norm_topk_prob"] = True
99839983
config["scoring_func"] = "softmax"
99849984

9985+
def set_vocab(self):
9986+
LlamaModel._set_vocab_mistral(self) # type: ignore
9987+
99859988
def set_gguf_parameters(self):
99869989
super().set_gguf_parameters()
99879990
MistralModel.set_mistral_config(self.gguf_writer, self.hparams)

0 commit comments

Comments
 (0)