diff --git a/convert-hf-to-gguf.py b/convert-hf-to-gguf.py index 8d7d75fb2..6d1d5620b 100755 --- a/convert-hf-to-gguf.py +++ b/convert-hf-to-gguf.py @@ -222,6 +222,8 @@ class Model: return NomicBertModel if model_architecture == "GemmaForCausalLM": return GemmaModel + if model_architecture == "Starcoder2ForCausalLM": + return Model return Model def _is_model_safetensors(self) -> bool: