methods/elasticdnn/model/__init__.py

from .base import ElasticDNNUtil

from .cnn import ElasticCNNUtil
from .vit import ElasticViTUtil