pfcfuse/mamba_ssm/__init__.py

6 lines
235 B
Python
Raw Permalink Normal View History

# __version__ = "1.2.0.post1"
#
# from mamba_ssm.ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn
# from mamba_ssm.modules.mamba_simple import Mamba
# from mamba_ssm.models.mixer_seq_simple import MambaLMHeadModel