Handle token_type_ids in LIMEForCausalLM

modeling_lime.py

@@ -72,6 +72,7 @@ class LIMEForCausalLM(PreTrainedModel, GenerationMixin):
             self,
             input_ids: torch.LongTensor,
             attention_mask: Optional[torch.Tensor] = None,
+            token_type_ids: Optional[torch.LongTensor] = None,
             labels: Optional[torch.LongTensor] = None,
             return_dict: Optional[bool] = None,
             **kwargs