src/transformers/models/mobilevit/modeling_tf_mobilevit.py/TFMobileViTModel/__init__
class TFMobileViTModel:    def __init__(self, config: MobileViTConfig, expand_output: bool = True, *inputs, **kwargs):
        super().__init__(config, *inputs, **kwargs)
        self.config = config
        self.expand_output = expand_output

        self.mobilevit = TFMobileViTMainLayer(config, expand_output=expand_output, name="mobilevit")