From 09210334bf84fd024fa3edfe32626fa27db83b89 Mon Sep 17 00:00:00 2001 From: Pierrick HYMBERT Date: Sat, 6 Apr 2024 16:00:32 +0200 Subject: [PATCH] model: dbrx fix python linter in convert-hf-to-gguf.py --- convert-hf-to-gguf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py index a847d118d..5d858a123 100755 --- a/convert-hf-to-gguf.py +++ b/convert-hf-to-gguf.py @@ -1446,7 +1446,6 @@ class Qwen2MoeModel(Model): def _set_vocab_gpt2(self): dir_model = self.dir_model - hparams = self.hparams tokens: list[str] = [] toktypes: list[int] = [] @@ -1478,6 +1477,7 @@ class Qwen2MoeModel(Model): special_vocab = gguf.SpecialVocab(dir_model, load_merges=True) special_vocab.add_to_gguf(self.gguf_writer) + @Model.register("MiniCPMForCausalLM") class MiniCPMModel(Model): model_arch = gguf.MODEL_ARCH.MINICPM