Update modeling_ltgbert.py

modeling_ltgbert.py

@@ -422,6 +422,7 @@ class LtgBertModel(LtgBertPreTrainedModel):
         output_hidden_states: Optional[bool] = None,
         output_attentions: Optional[bool] = None,
         return_dict: Optional[bool] = None,
+        token_type_ids=None
     ) -> Union[Tuple[torch.Tensor], BaseModelOutput]:
 
         output_attentions = output_attentions if output_attentions is not None else self.config.output_attentions
@@ -468,6 +469,7 @@ class LtgBertForMaskedLM(LtgBertModel):
         output_attentions: Optional[bool] = None,
         return_dict: Optional[bool] = None,
         labels: Optional[torch.LongTensor] = None,
+        token_type_ids=None
     ) -> Union[Tuple[torch.Tensor], MaskedLMOutput]:
         r"""
         labels (`torch.LongTensor` of shape `(batch_size, sequence_length)`, *optional*):