src/transformers/models/flaubert/modeling_tf_flaubert.py/TFFlaubertPredLayer/build
class TFFlaubertPredLayer:    def build(self, input_shape):
        # The output weights are the same as the input embeddings, but there is an output-only bias for each token.
        self.bias = self.add_weight(shape=(self.n_words,), initializer="zeros", trainable=True, name="bias")

        super().build(input_shape)