diff --git a/requirements/test.txt b/requirements/test.txt index 08ba964f22a44..93f00da1f101a 100644 --- a/requirements/test.txt +++ b/requirements/test.txt @@ -178,7 +178,6 @@ einops==0.8.1 # via # -r requirements/test.in # encodec - # mamba-ssm # terratorch # torchgeo # vector-quantize-pytorch @@ -418,8 +417,6 @@ lxml==5.3.0 # sacrebleu mako==1.3.10 # via alembic -mamba-ssm==2.2.5 - # via -r requirements/test.in markdown==3.8.2 # via mlflow markdown-it-py==3.0.0 @@ -476,8 +473,6 @@ networkx==3.2.1 # via # scikit-image # torch -ninja==1.11.1.3 - # via mamba-ssm nltk==3.9.1 # via rouge-score num2words==0.5.14 @@ -630,7 +625,6 @@ packaging==24.2 # lazy-loader # lightning # lightning-utilities - # mamba-ssm # matplotlib # mlflow-skinny # peft @@ -974,7 +968,6 @@ sentencepiece==0.2.0 setuptools==77.0.3 # via # lightning-utilities - # mamba-ssm # pytablewriter # torch # triton @@ -1086,7 +1079,6 @@ torch==2.7.1+cu128 # lightly # lightning # lm-eval - # mamba-ssm # mteb # open-clip-torch # peft @@ -1153,16 +1145,13 @@ transformers==4.55.0 # -r requirements/test.in # genai-perf # lm-eval - # mamba-ssm # peft # sentence-transformers # transformers-stream-generator transformers-stream-generator==0.0.5 # via -r requirements/test.in triton==3.3.1 - # via - # mamba-ssm - # torch + # via torch tritonclient==2.51.0 # via # -r requirements/test.in diff --git a/tests/models/language/generation/test_hybrid.py b/tests/models/language/generation/test_hybrid.py index 78d23c609fbee..584a3640eb92a 100644 --- a/tests/models/language/generation/test_hybrid.py +++ b/tests/models/language/generation/test_hybrid.py @@ -25,6 +25,7 @@ SSM_MODELS = [ HYBRID_MODELS = [ "ai21labs/Jamba-tiny-dev", + "pfnet/plamo-2-1b", "Zyphra/Zamba2-1.2B-instruct", "hmellor/tiny-random-BambaForCausalLM", "ibm-ai-platform/Bamba-9B-v1",