model: Add support for Nanbeige4.2 (#25994)

* support nanbeige4.2 model * fix * fix flake8 Lint check * fix loop bound check and drop redundant head_dim --------- Co-authored-by: root <lizongqiang@kanzhun.com>
2026-07-27 17:04:18 +02:00
parent 0324696b8e
commit b77d646751
10 changed files with 265 additions and 1 deletions
@@ -167,6 +167,7 @@ TEXT_MODEL_MAP: dict[str, str] = {
    "ModernBertForMaskedLM": "bert",
    "ModernBertForSequenceClassification": "bert",
    "ModernBertModel": "bert",
+    "NanbeigeForCausalLM": "nanbeige",
    "NemotronForCausalLM": "nemotron",
    "NemotronHForCausalLM": "nemotron",
    "NeoBERT": "bert",