File size: 341 Bytes
23d26f4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
__version__ = "2.2.4"
from .ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn
from .modules.mamba_simple import Mamba
from .modules.mamba2 import Mamba2
from .models.mixer_seq_simple import MambaLMHeadModel
__all__ = [
"selective_scan_fn",
"mamba_inner_fn",
"Mamba",
"Mamba2",
"MambaLMHeadModel",
]
|