File size: 404 Bytes
c0f89d0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | """
model — LLM architecture package.
Public API:
LLM : top-level decoder-only transformer/hybrid language model
LMConfig : configuration dataclass
Mamba2Block: Mamba-2 SSD block (used internally by LLM in hybrid mode)
"""
from .config import LMConfig
from .mamba_block import Mamba2Block
from .transformer import LLM
__all__ = [
"LLM",
"LMConfig",
"Mamba2Block",
]
|