model_files/configs/__init__.py

"""MiniMind Max2 Configuration Module"""
from .model_config import Max2Config, get_config, estimate_params, MAX2_CONFIGS

# Backward compatibility
Mind2Config = Max2Config
MIND2_CONFIGS = MAX2_CONFIGS

__all__ = [
    "Max2Config",
    "Mind2Config",
    "get_config",
    "estimate_params",
    "MAX2_CONFIGS",
    "MIND2_CONFIGS",
]