src/transformers/models/distilbert/modeling_tf_distilbert.py/TFDistilBertMainLayer/set_input_embeddings
class TFDistilBertMainLayer:    def set_input_embeddings(self, value):
        self.embeddings.weight = value
        self.embeddings.vocab_size = value.shape[0]