YOLO-Tutorial-v2/masked_image_modeling/models/vit/__init__.py

from .vit import build_vit
from .vit_mae import build_vit_mae
from .vit_cls import ViTForImageClassification