From c93821e97488076a794c10791e3a2657bbfc3965 Mon Sep 17 00:00:00 2001 From: Andrei Betlen Date: Thu, 27 Jun 2024 16:07:09 -0400 Subject: [PATCH] format fix --- 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 2770e3ca1..eff9b7a2d 100755 --- a/convert-hf-to-gguf.py +++ b/convert-hf-to-gguf.py @@ -2339,6 +2339,7 @@ class GemmaModel(Model): return [(self.map_tensor_name(name), data_torch)] + @Model.register("Gemma2ForCausalLM") class Gemma2Model(Model): model_arch = gguf.MODEL_ARCH.GEMMA2 @@ -2377,6 +2378,7 @@ class Gemma2Model(Model): return [(self.map_tensor_name(name), data_torch)] + @Model.register("Starcoder2ForCausalLM") class StarCoder2Model(Model): model_arch = gguf.MODEL_ARCH.STARCODER2