Skip to content

Commit 8cf6795

Browse files
committed
deps: install mamba_ssm from package instead of github
Signed-off-by: Anh Uong <[email protected]>
1 parent 6f25eed commit 8cf6795

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)