ColossalAI/colossalai/context/__init__.py

from .config import Config, ConfigException

# from .moe_context import MOE_CONTEXT

__all__ = [
    "Config",
    "ConfigException",
]