forked from state-spaces/mamba
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathpyproject.toml
More file actions
52 lines (49 loc) · 1.25 KB
/
pyproject.toml
File metadata and controls
52 lines (49 loc) · 1.25 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
[project]
name = "mamba_ssm"
description = "Mamba state-space model"
readme = "README.md"
authors = [
]
requires-python = ">= 3.9"
dynamic = ["version"]
license = { file = "LICENSE" } # Include a LICENSE file in your repo
keywords = ["cuda", "pytorch", "state-space model"]
classifiers = [
"Programming Language :: Python :: 3",
"License :: OSI Approved :: Apache Software License",
"Operating System :: Unix"
]
dependencies = [
"torch",
"tilelang==0.1.8",
"quack-kernels>=0.3.4",
"triton>=3.5.0",
"ninja",
"einops",
"transformers",
"packaging",
"setuptools>=61.0.0",
]
[project.urls]
Repository = "https://github.com/state-spaces/mamba"
[project.optional-dependencies]
causal-conv1d = [
"causal-conv1d>=1.2.0"
]
dev = [
"pytest"
]
[build-system]
# torch is intentionally excluded: pip's build isolation would install
# torch-cpu from PyPI, ignoring the user's CUDA-enabled torch.
# Users building from source should install torch first, then:
# pip install mamba-ssm --no-build-isolation
requires = [
"setuptools>=61.0.0",
"wheel",
"packaging",
"ninja",
]
build-backend = "setuptools.build_meta"