src/transformers/models/esm/modeling_esm.py/EsmOutput/forward
class EsmOutput:    def forward(self, hidden_states, input_tensor):
        hidden_states = self.dense(hidden_states)
        hidden_states = self.dropout(hidden_states)
        hidden_states += input_tensor
        return hidden_states