Update modeling_ltgbert.py

modeling_ltgbert.py

@@ -452,6 +452,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
@@ -498,6 +499,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*):