src/transformers/models/layoutlm/modeling_tf_layoutlm.py/TFLayoutLMLMPredictionHead/set_bias
class TFLayoutLMLMPredictionHead:    def set_bias(self, value: tf.Variable):
        self.bias = value["bias"]
        self.vocab_size = shape_list(value["bias"])[0]