Fix custom model loader token_type_ids compatibility

modeling_type_predictor.py

@@ -55,7 +55,10 @@ class NeoAraBERTTypePredictor(nn.Module):
         self.dropout = nn.Dropout(float(dropout))
         self.classifier = nn.Linear(self.hidden_size, int(num_labels))
 
-    def forward(self, input_ids=None, attention_mask=None, labels=None):
+    def forward(self, input_ids=None, attention_mask=None, token_type_ids=None, labels=None, **kwargs):
+        # NeoAraBERT was trained here with input_ids and attention_mask only.
+        # Accept token_type_ids/extra tokenizer keys so callers can pass
+        # tokenizer(..., return_tensors="pt") directly.
         outputs = self.encoder(input_ids=input_ids, attention_mask=attention_mask)
         hidden = outputs.last_hidden_state if hasattr(outputs, "last_hidden_state") else outputs[0]
         logits = self.classifier(self.dropout(hidden[:, 0, :]))

release_manifest.json

@@ -3,6 +3,10 @@
     "sha256": "930dd95e0feefb5e1ff497f2c60930b86f3eb318179851a1ffd7f61d48b4d780",
     "bytes": 13809
   },
+  "__pycache__/modeling_type_predictor.cpython-312.pyc": {
+    "sha256": "d3fccd58c9d18ff94f5949426ee4b0d66a1e2de8961410d178d44d19c3e73d8c",
+    "bytes": 6336
+  },
   "checkpoint_summary.json": {
     "sha256": "eab687212a3bd05b0659ff7a255dc552b7d5bca87f75e5beffd318e5f384bf2b",
     "bytes": 570
@@ -200,8 +204,8 @@
     "bytes": 2994
   },
   "modeling_type_predictor.py": {
-    "sha256": "41a6c1b1ac7615622b8959fb7bf0561be1930b4b63a6346ba89eec89d40baef8",
-    "bytes": 4316
+    "sha256": "26b66582aeebe272a0a6136c28ec4b215f4144854732fe48dd83d93d7b9315c6",
+    "bytes": 4554
   },
   "predictions/test_predictions.jsonl": {
     "sha256": "fcf7973d1e70244cde5b8639296e0150392d176f3150037dadfe01ddeaa88a54",