summaryrefslogtreecommitdiff
path: root/convert_hf_to_gguf.py
diff options
context:
space:
mode:
authorsaood06 <saood05@gmail.com>2025-04-22 01:34:13 -0500
committerGitHub <noreply@github.com>2025-04-22 08:34:13 +0200
commitcc398007238cbbb064609cbdc9bc4aab03c658d7 (patch)
treea414370696faa5ad8a89ac5aa7ec7cf2171a78a4 /convert_hf_to_gguf.py
parent93cd77b65501246603061c7ee2801a992e3c6312 (diff)
Add support for bitnet2b_2501 model (#337)
* add support for bitnet2b_2501 model * Fixes * Support both model names --------- Co-authored-by: potassiummmm <zhou.hansong@outlook.com>
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