src/transformers/models/distilbert/modeling_tf_distilbert.py/TFDistilBertLMHead/__init__
class TFDistilBertLMHead:    def __init__(self, config, input_embeddings, **kwargs):
        super().__init__(**kwargs)

        self.vocab_size = config.vocab_size
        self.dim = config.dim

        # The output weights are the same as the input embeddings, but there is
        # an output-only bias for each token.
        self.input_embeddings = input_embeddings