conversion: accept BitNetForCausalLM architecture name (#25769)

Microsoft BitNet Hugging Face configs use BitNetForCausalLM while the
converter only registered BitnetForCausalLM, causing conversion to fail
with "Model BitNetForCausalLM is not supported".

Register both spellings in TEXT_MODEL_MAP and the Bitnet model class.

Fixes ggml-org/llama.cpp#25629

diff --git a/conversion/__init__.py b/conversion/__init__.py
index 02102fac8d90053baa7605d077d6c37b06272933..465c010f667a08379350f9995e89473e162e2066 100644 (file)
--- a/conversion/__init__.py
+++ b/conversion/__init__.py
@@ -31,6 +31,7 @@ TEXT_MODEL_MAP: dict[str, str] = {
     "BertForSequenceClassification": "bert",
     "BertModel": "bert",
     "BitnetForCausalLM": "bitnet",
+    "BitNetForCausalLM": "bitnet",
     "BloomForCausalLM": "bloom",
     "BloomModel": "bloom",
     "CamembertModel": "bert",

diff --git a/conversion/bitnet.py b/conversion/bitnet.py
index a66446abee2f0dc386cadb4e870403f972841ead..0c2baee876081a399b098db9f2976275c10bbc43 100644 (file)
--- a/conversion/bitnet.py
+++ b/conversion/bitnet.py
@@ -8,7 +8,7 @@ if TYPE_CHECKING:
 from .base import ModelBase, TextModel, gguf
 
 
-@ModelBase.register("BitnetForCausalLM")
+@ModelBase.register("BitnetForCausalLM", "BitNetForCausalLM")
 class BitnetModel(TextModel):
     model_arch = gguf.MODEL_ARCH.BITNET