From b7cdf7b1cc580a76f93ecd6e3500a958eafabb9a Mon Sep 17 00:00:00 2001 From: Tri Dao Date: Fri, 6 Dec 2024 00:40:51 -0800 Subject: [PATCH] Add back info in setup.py to upload to PyPI --- mamba_ssm/__init__.py | 2 +- setup.py | 22 ++++++++++++++++++++-- 2 files changed, 21 insertions(+), 3 deletions(-) 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", + ], )