From 3d7206e6eaadb3ed9142d762e9a17fa39d226343 Mon Sep 17 00:00:00 2001 From: Saood Karim Date: Mon, 21 Apr 2025 21:47:17 -0500 Subject: [PATCH] Support both model names --- convert_hf_to_gguf.py | 1 + 1 file changed, 1 insertion(+) 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