File size: 528 Bytes
85c0263 3f2a020 85c0263 8ce5a49 85c0263 8ce5a49 85c0263 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
__version__ = "2.2.4"
from .ops import selective_state_update, mamba_chunk_scan_combined, mamba_split_conv1d_scan_combined
from .ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn
from .modules import Mamba, Mamba2
from .models import MambaLMHeadModel
falcon_mamba_inner_fn = mamba_inner_fn
__all__ = [
"selective_scan_fn", "mamba_inner_fn", "falcon_mamba_inner_fn", "selective_state_update", "mamba_chunk_scan_combined", "mamba_split_conv1d_scan_combined",
"Mamba", "Mamba2", "MambaLMHeadModel",
]
|