src/transformers/models/bert_generation/modeling_bert_generation.py/BertGenerationDecoder/set_output_embeddings
class BertGenerationDecoder:    def set_output_embeddings(self, new_embeddings):
        self.lm_head.decoder = new_embeddings