Allow conversion of Mistral HF models
This commit is contained in:
parent
d199ca79f2
commit
77f313e31b
1 changed files with 8 additions and 0 deletions
|
@ -1051,6 +1051,14 @@ class MixtralModel(Model):
|
|||
self._set_vocab_sentencepiece()
|
||||
|
||||
|
||||
@Model.register("MistralForCausalLM")
|
||||
class MistralModel(Model):
|
||||
model_arch = gguf.MODEL_ARCH.LLAMA
|
||||
|
||||
def set_vocab(self):
|
||||
self._set_vocab_sentencepiece()
|
||||
|
||||
|
||||
@Model.register("MiniCPMForCausalLM")
|
||||
class MiniCPMModel(Model):
|
||||
model_arch = gguf.MODEL_ARCH.MINICPM
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue