VBench/vbench/third_party/umt/models/__init__.py

from .clip import clip_b16, clip_l14, clip_l14_336
# from .modeling_finetune import vit_base_patch16_224, vit_base_patch16_384, vit_large_patch16_224, vit_large_patch16_384
from .modeling_finetune import vit_large_patch16_224
from .modeling_pretrain_umt import pretrain_umt_base_patch16_224, pretrain_umt_large_patch16_224 
from .modeling_pretrain import pretrain_videomae_base_patch16_224, pretrain_videomae_large_patch16_224, pretrain_videomae_huge_patch16_224