Add kwargs to T5ForClassification forward

turkish_lm_tuner/t5_classifier.py

@@ -39,7 +39,7 @@ def __init__(self, pretrained_model_name, config, num_labels, problem_type, drop
 
         self.model_parallel = False
 
-    def forward(self, input_ids, attention_mask=None, labels=None):
+    def forward(self, input_ids, attention_mask=None, labels=None, **kwargs):
         encoder_output = self.encoder(input_ids, attention_mask=attention_mask)
         if self.config.problem_type == "token_classification":
             sequence_output = encoder_output.last_hidden_state