def METHOD_NAME(arch='vit_b16_224', pretrained=None):
    """ build a ViT + ViM model
        Args:
            arch: name of backbone
            pretrained: weights of pretrained model
    """
    model_args = model_dict[arch]
    model = VisionTransformer(**model_args)
    model.load_state_dict(pretrained)
    return model