src/transformers/models/bert_generation/modeling_bert_generation.py/BertGenerationOnlyLMHead/forward
class BertGenerationOnlyLMHead:    def forward(self, hidden_states):
        logits = self.decoder(hidden_states)
        return logits