src/transformers/models/rembert/modeling_tf_rembert.py/TFRemBertLMPredictionHead/set_bias
class TFRemBertLMPredictionHead:    def set_bias(self, value: tf.Variable):
        self.decoder_bias = value["decoder_bias"]
        self.config.vocab_size = shape_list(value["decoder_bias"])[0]