From 1f6929e5571ef9bdf4947382740d03d5b533d971 Mon Sep 17 00:00:00 2001 From: Ashish <1856117+ashishdatta@users.noreply.github.com> Date: Sun, 14 Apr 2024 16:46:26 -0700 Subject: [PATCH] Flake8 format --- convert-hf-to-gguf.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py index 3a73cce97..120918369 100755 --- a/convert-hf-to-gguf.py +++ b/convert-hf-to-gguf.py @@ -1186,6 +1186,7 @@ class PersimmonModel(Model): @Model.register("StableLmForCausalLM", "StableLMEpochForCausalLM", "LlavaStableLMEpochForCausalLM") class StableLMModel(Model): model_arch = gguf.MODEL_ARCH.STABLELM + def set_vocab(self): if (self.dir_model / "tokenizer.json").is_file(): self._set_vocab_gpt2() @@ -1290,6 +1291,7 @@ class StableLMModel(Model): self.gguf_writer.add_tensor(new_name, data) + @Model.register("LlamaForCausalLM", "MistralForCausalLM", "MixtralForCausalLM") class LlamaModel(Model): model_arch = gguf.MODEL_ARCH.LLAMA