Update modeling_ltgbert.py
Browse files- 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,
|