src/transformers/models/groupvit/modeling_tf_groupvit.py/TFGroupViTVisionMainLayer/__init__
class TFGroupViTVisionMainLayer:    def __init__(self, config: GroupViTVisionConfig, **kwargs):
        super().__init__(**kwargs)
        self.config = config
        self.vision_model = TFGroupViTVisionTransformer(config, name="vision_model")