| """ | |
| MiniMind Max2 Model Package | |
| A lightweight, efficient language model designed for edge deployment. | |
| """ | |
| from .mind2_model import ( | |
| Max2ForCausalLM, | |
| Max2Model, | |
| Mind2ForCausalLM, | |
| Mind2Model, | |
| create_model | |
| ) | |
| from .components import ( | |
| Max2Attention, | |
| Max2MoE, | |
| Max2DecoderLayer, | |
| Max2RMSNorm, | |
| Max2RotaryEmbedding, | |
| Max2MLP, | |
| Max2Expert, | |
| # Backward compatibility | |
| Mind2Attention, | |
| Mind2MoE, | |
| Mind2DecoderLayer, | |
| Mind2RMSNorm, | |
| Mind2RotaryEmbedding, | |
| ) | |
| __all__ = [ | |
| # Max2 (primary) | |
| "Max2ForCausalLM", | |
| "Max2Model", | |
| "Max2Attention", | |
| "Max2MoE", | |
| "Max2DecoderLayer", | |
| "Max2RMSNorm", | |
| "Max2RotaryEmbedding", | |
| "Max2MLP", | |
| "Max2Expert", | |
| # Mind2 (backward compatibility) | |
| "Mind2ForCausalLM", | |
| "Mind2Model", | |
| "Mind2Attention", | |
| "Mind2MoE", | |
| "Mind2DecoderLayer", | |
| "Mind2RMSNorm", | |
| "Mind2RotaryEmbedding", | |
| # Factory | |
| "create_model", | |
| ] | |
| __version__ = "1.0.0" | |