Upload modeling_esm_plusplus.py with huggingface_hub
Browse files- modeling_esm_plusplus.py +1 -0
modeling_esm_plusplus.py
CHANGED
@@ -950,6 +950,7 @@ class ESMplusplusForSequenceClassification(ESMplusplusForMaskedLM, EmbeddingMixi
|
|
950 |
output_attentions: Optional[bool] = None,
|
951 |
output_hidden_states: Optional[bool] = None,
|
952 |
return_dict: Optional[bool] = None, # to play nice with HF adjacent packages
|
|
|
953 |
) -> ESMplusplusOutput:
|
954 |
"""Forward pass for sequence classification.
|
955 |
|
|
|
950 |
output_attentions: Optional[bool] = None,
|
951 |
output_hidden_states: Optional[bool] = None,
|
952 |
return_dict: Optional[bool] = None, # to play nice with HF adjacent packages
|
953 |
+
**kwargs,
|
954 |
) -> ESMplusplusOutput:
|
955 |
"""Forward pass for sequence classification.
|
956 |
|