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>
This commit is contained in:
saood06
2025-04-22 01:34:13 -05:00
committed by GitHub
parent 16f945d9bb
commit e6c85a5b95
4 changed files with 330 additions and 1 deletions

View File

@@ -1598,6 +1598,7 @@ class LlamaModel(Model):
@Model.register("BitnetForCausalLM")
@Model.register("BitNetForCausalLM")
class BitnetModel(Model):
model_arch = gguf.MODEL_ARCH.BITNET