Skip to content

Commit 7db3dff

Browse files
committed
update mamba_ssm to 2.2.5
Signed-off-by: elvischenv <219235043+elvischenv@users.noreply.github.com>
1 parent b77c7d3 commit 7db3dff

File tree

2 files changed

+25
-5
lines changed

2 files changed

+25
-5
lines changed

requirements/test.in

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ torch==2.7.1
2626
torchaudio==2.7.1
2727
torchvision==0.22.1
2828
transformers_stream_generator # required for qwen-vl test
29-
mamba_ssm # required for plamo2 test
29+
mamba_ssm==2.2.5 # required for plamo2 test
3030
matplotlib # required for qwen-vl test
3131
mistral_common[opencv] >= 1.8.0 # required for voxtral test
3232
num2words # required for smolvlm test

requirements/test.txt

Lines changed: 24 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,10 @@ argcomplete==3.5.1
3131
# via datamodel-code-generator
3232
arrow==1.3.0
3333
# via isoduration
34+
async-timeout==5.0.1
35+
# via
36+
# aiohttp
37+
# redis
3438
attrs==24.2.0
3539
# via
3640
# aiohttp
@@ -141,6 +145,11 @@ eval-type-backport==0.2.2
141145
# via mteb
142146
evaluate==0.4.3
143147
# via lm-eval
148+
exceptiongroup==1.3.0
149+
# via
150+
# anyio
151+
# hypothesis
152+
# pytest
144153
fastparquet==2024.11.0
145154
# via genai-perf
146155
fastrlock==0.8.2
@@ -291,7 +300,7 @@ lxml==5.3.0
291300
# via
292301
# blobfile
293302
# sacrebleu
294-
mamba-ssm==2.2.4
303+
mamba-ssm==2.2.5
295304
# via -r requirements/test.in
296305
markdown-it-py==3.0.0
297306
# via rich
@@ -702,7 +711,6 @@ setuptools==77.0.3
702711
# via
703712
# mamba-ssm
704713
# pytablewriter
705-
# torch
706714
# triton
707715
shellingham==1.5.4
708716
# via typer
@@ -767,8 +775,13 @@ tokenizers==0.21.1
767775
# via
768776
# -r requirements/test.in
769777
# transformers
778+
toml==0.10.2
779+
# via datamodel-code-generator
770780
tomli==2.2.1
771-
# via schemathesis
781+
# via
782+
# black
783+
# pytest
784+
# schemathesis
772785
tomli-w==1.2.0
773786
# via schemathesis
774787
torch==2.7.1+cu128
@@ -829,7 +842,9 @@ transformers==4.53.2
829842
transformers-stream-generator==0.0.5
830843
# via -r requirements/test.in
831844
triton==3.3.1
832-
# via torch
845+
# via
846+
# mamba-ssm
847+
# torch
833848
tritonclient==2.51.0
834849
# via
835850
# -r requirements/test.in
@@ -845,14 +860,19 @@ types-python-dateutil==2.9.0.20241206
845860
# via arrow
846861
typing-extensions==4.12.2
847862
# via
863+
# anyio
864+
# black
865+
# exceptiongroup
848866
# huggingface-hub
849867
# librosa
850868
# mistral-common
851869
# mteb
870+
# multidict
852871
# pqdm
853872
# pydantic
854873
# pydantic-core
855874
# pydantic-extra-types
875+
# rich
856876
# torch
857877
# typer
858878
# typing-inspection

0 commit comments

Comments
 (0)