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