Add back kwargs to CaduceusForSequenceClassification forward
Browse files- modeling_caduceus.py +1 -0
modeling_caduceus.py
CHANGED
|
@@ -543,6 +543,7 @@ class CaduceusForSequenceClassification(CaduceusPreTrainedModel):
|
|
| 543 |
labels: Optional[torch.LongTensor] = None,
|
| 544 |
output_hidden_states: Optional[bool] = None,
|
| 545 |
return_dict: Optional[bool] = None,
|
|
|
|
| 546 |
) -> Union[Tuple, SequenceClassifierOutput]:
|
| 547 |
r"""
|
| 548 |
labels (`torch.LongTensor` of shape `(batch_size,)`, *optional*):
|
|
|
|
| 543 |
labels: Optional[torch.LongTensor] = None,
|
| 544 |
output_hidden_states: Optional[bool] = None,
|
| 545 |
return_dict: Optional[bool] = None,
|
| 546 |
+
**kwargs,
|
| 547 |
) -> Union[Tuple, SequenceClassifierOutput]:
|
| 548 |
r"""
|
| 549 |
labels (`torch.LongTensor` of shape `(batch_size,)`, *optional*):
|