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 |
|