from .clip_models import ClipModel


VALID_NAMES = {
    'CLIP:ViT-B/16_svd': 'openai/clip-vit-base-patch16',
    'CLIP:ViT-B/32_svd': 'openai/clip-vit-base-patch32',
    'CLIP:ViT-L/14_svd': 'openai/clip-vit-large-patch14',
    'SigLIP:ViT-L/16_256_svd': 'google/siglip-large-patch16-256',
    'BEiTv2:ViT-L/16_svd': 'microsoft/beitv2-large-patch16-224',
}
def get_model(name, opt):
    assert name in VALID_NAMES.keys()
    if name.startswith("CLIP:"):
        return ClipModel(VALID_NAMES[name], opt)
    else:
        assert False