src/transformers/models/flaubert/modeling_tf_flaubert.py/TFFlaubertMainLayer/set_input_embeddings
class TFFlaubertMainLayer:    def set_input_embeddings(self, value):
        self.embeddings.weight = value
        self.embeddings.vocab_size = shape_list(value)[0]