GeneMamba / setup.py
mineself2016's picture
Upload GeneMamba model
54cd552 verified
"""
Setup script for GeneMamba Hugging Face package.
"""
from setuptools import setup, find_packages
with open("README.md", "r", encoding="utf-8") as fh:
long_description = fh.read()
setup(
name="genemamba-hf",
version="0.1.0",
author="GeneMamba Contributors",
description="GeneMamba: Foundation model for single-cell analysis on Hugging Face",
long_description=long_description,
long_description_content_type="text/markdown",
url="https://huggingface.co/models?search=genemamba",
packages=find_packages(),
classifiers=[
"Development Status :: 3 - Alpha",
"Intended Audience :: Science/Research",
"Topic :: Scientific/Engineering :: Bio-Informatics",
"License :: OSI Approved :: Apache Software License",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.9",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
],
python_requires=">=3.9",
install_requires=[
"torch>=2.0.0",
"transformers>=4.40.0",
"mamba-ssm>=2.2.0",
"tokenizers>=0.19.0",
],
)