src/transformers/models/bert/modeling_tf_bert.py/TFBertLMPredictionHead/build
class TFBertLMPredictionHead:    def build(self, input_shape: tf.TensorShape):
        self.bias = self.add_weight(shape=(self.vocab_size,), initializer="zeros", trainable=True, name="bias")

        super().build(input_shape)