src/transformers/models/convbert/modeling_convbert.py/ConvBertForMaskedLM/set_output_embeddings
class ConvBertForMaskedLM:    def set_output_embeddings(self, word_embeddings):
        self.generator_lm_head = word_embeddings