protify/base_models/__init__.py

from .get_base_models import (
    get_base_model,
    get_base_model_for_training,
    get_tokenizer,
    experimental_models,
    BaseModelArguments
)

try:
    from .model_descriptions import model_descriptions
except ImportError:
    model_descriptions = {}