src/transformers/models/roberta_prelayernorm/modeling_tf_roberta_prelayernorm.py/TFRobertaPreLayerNormLMHead/set_bias
class TFRobertaPreLayerNormLMHead:    def set_bias(self, value):
        self.bias = value["bias"]
        self.config.vocab_size = shape_list(value["bias"])[0]