src/transformers/models/roberta_prelayernorm/modeling_tf_roberta_prelayernorm.py/TFRobertaPreLayerNormLMHead/set_output_embeddings
class TFRobertaPreLayerNormLMHead:    def set_output_embeddings(self, value):
        self.decoder.weight = value
        self.decoder.vocab_size = shape_list(value)[0]