src/transformers/models/groupvit/modeling_tf_groupvit.py/TFGroupViTTextMainLayer/set_input_embeddings
class TFGroupViTTextMainLayer:    def set_input_embeddings(self, value: tf.Variable):
        self.text_model.embeddings.weight = value
        self.text_model.embeddings.vocab_size = shape_list(value)[0]