Skip to content

Commit 456fe2a

Browse files
authored
Merge pull request #508 from anhuong/fix-mamba-dep-release
build(deps): install mamba_ssm from package instead of github
2 parents 6f25eed + 8cf6795 commit 456fe2a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,7 @@ aim = ["aim>=3.19.0,<4.0"]
4848
mlflow = ["mlflow"]
4949
fms-accel = ["fms-acceleration>=0.6"]
5050
gptq-dev = ["auto_gptq>0.4.2", "optimum>=1.15.0"]
51-
mamba = ["mamba_ssm[causal-conv1d] @ git+https://github.com/state-spaces/mamba.git"]
51+
mamba = ["mamba_ssm[causal-conv1d]>=2.0.0,<3.0.0"]
5252
scanner-dev = ["HFResourceScanner>=0.1.0"]
5353

5454

0 commit comments

Comments
 (0)