spectre/models/__init__.py

from .vision_transformer import (
    VisionTransformer,
    vit_tiny_patch16_128, 
    vit_small_patch16_128, 
    vit_base_patch16_128, 
    vit_base_patch32_128,
    vit_large_patch16_128,
    vit_large_patch32_128,
)
from .vision_transformer_features import (
    FeatureVisionTransformer,
    feat_vit_tiny,
    feat_vit_small,
    feat_vit_base,
    feat_vit_large,
)

__all__ = [
    'VisionTransformer',
    'vit_tiny_patch16_128',
    'vit_small_patch16_128',
    'vit_base_patch16_128',
    'vit_base_patch32_128',
    'vit_large_patch16_128',
    'vit_large_patch32_128',
    'FeatureVisionTransformer',
    'feat_vit_tiny',
    'feat_vit_small',
    'feat_vit_base',
    'feat_vit_large',
]