diff --git a/mamba_ssm/__init__.py b/mamba_ssm/__init__.py index d3e769b2..97e4ea8e 100644 --- a/mamba_ssm/__init__.py +++ b/mamba_ssm/__init__.py @@ -1,4 +1,4 @@ -__version__ = "2.2.3" +__version__ = "2.2.3.post1" from mamba_ssm.ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn from mamba_ssm.modules.mamba_simple import Mamba diff --git a/setup.py b/setup.py index dfc593a3..4cb95970 100755 --- a/setup.py +++ b/setup.py @@ -348,13 +348,31 @@ def run(self): "mamba_ssm.egg-info", ) ), + author="Tri Dao, Albert Gu", + author_email="tri@tridao.me, agu@cs.cmu.edu", + description="Mamba state-space model", long_description=long_description, long_description_content_type="text/markdown", - + url="https://github.com/state-spaces/mamba", + classifiers=[ + "Programming Language :: Python :: 3", + "License :: OSI Approved :: BSD License", + "Operating System :: Unix", + ], ext_modules=ext_modules, cmdclass={"bdist_wheel": CachedWheelsCommand, "build_ext": BuildExtension} if ext_modules else { "bdist_wheel": CachedWheelsCommand, - } + }, + python_requires=">=3.9", + install_requires=[ + "torch", + "packaging", + "ninja", + "einops", + "triton", + "transformers", + # "causal_conv1d>=1.4.0", + ], )