Minor fix for correct input forwarding

model.py

@@ -323,13 +323,13 @@ class NeoBERTLMHead(NeoBERTPreTrainedModel):
     ):
 
         output = self.model.forward(
-            input_ids,
-            position_ids,
-            max_seqlen,
-            cu_seqlens,
-            attention_mask,
-            output_hidden_states,
-            output_attentions,
+            input_ids=input_ids,
+            position_ids=position_ids,
+            max_seqlen=max_seqlen,
+            cu_seqlens=cu_seqlens,
+            attention_mask=attention_mask,
+            output_hidden_states=output_hidden_states,
+            output_attentions=output_attentions,
         )
         logits = self.decoder(output.last_hidden_state)
 
@@ -380,13 +380,13 @@ class NeoBERTForSequenceClassification(NeoBERTPreTrainedModel):
     ):
 
         output = self.model.forward(
-            input_ids,
-            position_ids,
-            max_seqlen,
-            cu_seqlens,
-            attention_mask,
-            output_hidden_states,
-            output_attentions,
+            input_ids=input_ids,
+            position_ids=position_ids,
+            max_seqlen=max_seqlen,
+            cu_seqlens=cu_seqlens,
+            attention_mask=attention_mask,
+            output_hidden_states=output_hidden_states,
+            output_attentions=output_attentions,
         )
         hidden_states = output.last_hidden_state