lhallee commited on
Commit
3e5d77a
·
verified ·
1 Parent(s): 6121b72

Upload modeling_fastesm.py with huggingface_hub

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