inference/models/vit/__init__.py

from inference.models.vit.vit_classification import VitClassification