diff options
author | saood06 <saood05@gmail.com> | 2025-04-22 01:34:13 -0500 |
---|---|---|
committer | GitHub <noreply@github.com> | 2025-04-22 08:34:13 +0200 |
commit | cc398007238cbbb064609cbdc9bc4aab03c658d7 (patch) | |
tree | a414370696faa5ad8a89ac5aa7ec7cf2171a78a4 /convert_hf_to_gguf.py | |
parent | 93cd77b65501246603061c7ee2801a992e3c6312 (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-x | convert_hf_to_gguf.py | 1 |
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 |