diff --git a/convert_hf_to_gguf.py b/convert_hf_to_gguf.py index 02ff8e3..05855aa 100755 --- a/convert_hf_to_gguf.py +++ b/convert_hf_to_gguf.py @@ -1516,7 +1516,7 @@ def prepare_tensors(self): raise ValueError(f"Unprocessed norms: {norms}") -@Model.register("LLaMAForCausalLM", "LlamaForCausalLM", "MistralForCausalLM", "MixtralForCausalLM", "DeciLMCausalModel") +@Model.register("LLaMAForCausalLM", "LlamaForCausalLM", "MistralForCausalLM", "MixtralForCausalLM", "DeciLMForCausalLM") class LlamaModel(Model): model_arch = gguf.MODEL_ARCH.LLAMA