Update model.py

Add default value to return_dict if not passed.

model.py

@@ -378,7 +378,8 @@ class NeoBERTForSequenceClassification(NeoBERTPreTrainedModel):
         labels: Optional[torch.Tensor] = None,
         return_dict: Optional[bool] = None,
     ):
-
+        return_dict = return_dict if return_dict is not None else self.config.use_return_dict
+        
         output = self.model.forward(
             input_ids=input_ids,
             position_ids=position_ids,