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