Support both model names

This commit is contained in:
Saood Karim
2025-04-21 21:47:17 -05:00
parent 2641658c90
commit 3d7206e6ea

View File

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