yairschiff commited on
Commit
22386c3
·
verified ·
1 Parent(s): e19d855

Add back kwargs to CaduceusForSequenceClassification forward

Browse files
Files changed (1) hide show
  1. 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*):