Upload modeling_fastesm.py with huggingface_hub
Browse files- modeling_fastesm.py +1 -0
modeling_fastesm.py
CHANGED
|
@@ -824,6 +824,7 @@ class FastEsmForMaskedLM(FastEsmPreTrainedModel):
|
|
| 824 |
return EsmMaskedLMOutput(
|
| 825 |
loss=loss,
|
| 826 |
logits=prediction_scores,
|
|
|
|
| 827 |
hidden_states=outputs.hidden_states,
|
| 828 |
attentions=outputs.attentions,
|
| 829 |
)
|
|
|
|
| 824 |
return EsmMaskedLMOutput(
|
| 825 |
loss=loss,
|
| 826 |
logits=prediction_scores,
|
| 827 |
+
last_hidden_state=sequence_output,
|
| 828 |
hidden_states=outputs.hidden_states,
|
| 829 |
attentions=outputs.attentions,
|
| 830 |
)
|