diff --git a/convert_hf_to_gguf.py b/convert_hf_to_gguf.py index 1ee82724..a6ab09c0 100755 --- a/convert_hf_to_gguf.py +++ b/convert_hf_to_gguf.py @@ -1598,6 +1598,7 @@ class LlamaModel(Model): @Model.register("BitnetForCausalLM") +@Model.register("BitNetForCausalLM") class BitnetModel(Model): model_arch = gguf.MODEL_ARCH.BITNET