amroadel1 commited on
Commit
64b17ca
1 Parent(s): dd398b7

Update modeling_ltgbert.py

Browse files
Files changed (1) hide show
  1. modeling_ltgbert.py +1 -0
modeling_ltgbert.py CHANGED
@@ -524,6 +524,7 @@ class LtgBertForSequenceClassification(LtgBertModel):
524
  self,
525
  input_ids: Optional[torch.Tensor] = None,
526
  attention_mask: Optional[torch.Tensor] = None,
 
527
  output_attentions: Optional[bool] = None,
528
  output_hidden_states: Optional[bool] = None,
529
  return_dict: Optional[bool] = None,
 
524
  self,
525
  input_ids: Optional[torch.Tensor] = None,
526
  attention_mask: Optional[torch.Tensor] = None,
527
+ token_type_ids: Optional[torch.Tensor] = None,
528
  output_attentions: Optional[bool] = None,
529
  output_hidden_states: Optional[bool] = None,
530
  return_dict: Optional[bool] = None,