From 0044bdf410d06debc55bcb66fbe626c29d15659d Mon Sep 17 00:00:00 2001 From: Eli Uriegas Date: Tue, 24 Jun 2025 17:00:08 -0700 Subject: [PATCH 1/2] requirements: Remove mamba_ssm from test.txt This dependency wasn't actually getting used and caused issues once we updated torch to 2.7.0. Signed-off-by: Eli Uriegas --- requirements/test.in | 3 +-- requirements/test.txt | 9 --------- 2 files changed, 1 insertion(+), 11 deletions(-) diff --git a/requirements/test.in b/requirements/test.in index e8f44059fcf8..444e88e25f93 100644 --- a/requirements/test.in +++ b/requirements/test.in @@ -26,7 +26,6 @@ torch==2.7.0 torchaudio==2.7.0 torchvision==0.22.0 transformers_stream_generator # required for qwen-vl test -mamba_ssm # required for plamo2 test matplotlib # required for qwen-vl test mistral_common[opencv] >= 1.6.2 # required for pixtral test num2words # required for smolvlm test @@ -51,4 +50,4 @@ numpy runai-model-streamer==0.11.0 runai-model-streamer-s3==0.11.0 fastsafetensors>=0.1.10 -pydantic>=2.10 # 2.9 leads to error on python 3.10 \ No newline at end of file +pydantic>=2.10 # 2.9 leads to error on python 3.10 diff --git a/requirements/test.txt b/requirements/test.txt index 16d8ee54adcf..546cf622bc09 100644 --- a/requirements/test.txt +++ b/requirements/test.txt @@ -128,7 +128,6 @@ einops==0.8.0 # via # -r requirements/test.in # encodec - # mamba-ssm # vector-quantize-pytorch # vocos einx==0.3.0 @@ -288,8 +287,6 @@ lxml==5.3.0 # via # blobfile # sacrebleu -mamba-ssm==2.2.4 - # via -r requirements/test.in markdown-it-py==3.0.0 # via rich markupsafe==3.0.1 @@ -329,8 +326,6 @@ mypy-extensions==1.0.0 # via black networkx==3.2.1 # via torch -ninja==1.11.1.3 - # via mamba-ssm nltk==3.9.1 # via rouge-score num2words==0.5.14 @@ -432,7 +427,6 @@ packaging==24.2 # fastparquet # huggingface-hub # lazy-loader - # mamba-ssm # matplotlib # peft # plotly @@ -688,7 +682,6 @@ sentencepiece==0.2.0 # via mistral-common setuptools==77.0.3 # via - # mamba-ssm # pytablewriter # torch # triton @@ -765,7 +758,6 @@ torch==2.7.0+cu128 # encodec # fastsafetensors # lm-eval - # mamba-ssm # mteb # peft # runai-model-streamer @@ -805,7 +797,6 @@ transformers==4.52.4 # -r requirements/test.in # genai-perf # lm-eval - # mamba-ssm # peft # sentence-transformers # transformers-stream-generator From 9bd1ad54d6f0cbb8404bfe38bb2e445953a6de00 Mon Sep 17 00:00:00 2001 From: Eli Uriegas Date: Tue, 24 Jun 2025 17:06:33 -0700 Subject: [PATCH 2/2] also remove from dockerfile Signed-off-by: Eli Uriegas --- docker/Dockerfile | 10 +--------- 1 file changed, 1 insertion(+), 9 deletions(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index cf9c245a9517..fe19e9653c15 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -167,10 +167,6 @@ FROM base as dev ENV UV_HTTP_TIMEOUT=500 ENV UV_INDEX_STRATEGY="unsafe-best-match" -# Workaround for #17068 -RUN --mount=type=cache,target=/root/.cache/uv \ - uv pip install --system --no-build-isolation "git+https://github.com/state-spaces/mamba@v2.2.4" - COPY requirements/lint.txt requirements/lint.txt COPY requirements/test.txt requirements/test.txt COPY requirements/dev.txt requirements/dev.txt @@ -302,12 +298,8 @@ ADD . /vllm-workspace/ ENV UV_HTTP_TIMEOUT=500 ENV UV_INDEX_STRATEGY="unsafe-best-match" -# Workaround for #17068 -RUN --mount=type=cache,target=/root/.cache/uv \ - uv pip install --system --no-build-isolation "git+https://github.com/state-spaces/mamba@v2.2.4" - # install development dependencies (for testing) -RUN --mount=type=cache,target=/root/.cache/uv \ +RUN --mount=type=cache,target=/root/.cache/uv \ CUDA_MAJOR="${CUDA_VERSION%%.*}"; \ if [ "$CUDA_MAJOR" -ge 12 ]; then \ uv pip install --system -r requirements/dev.txt; \