DA-2-repo/src/da2/model/__init__.py

from .spherevit import (
    SphereViT
)
from .vit_w_esphere import (
    ViT_w_Esphere
)

__all__ = [
    'SphereViT',
    'ViT_w_Esphere',
]