File size: 972 Bytes
8b187bb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
"""
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"
|