summaryrefslogtreecommitdiff
path: root/convert_hf_to_gguf.py
diff options
context:
space:
mode:
Diffstat (limited to 'convert_hf_to_gguf.py')
-rwxr-xr-xconvert_hf_to_gguf.py1
1 files changed, 1 insertions, 0 deletions
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