xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-05-23 02:44:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Fix] Update mamba_ssm to 2.2.5 (#21421)

Browse Source 
Signed-off-by: elvischenv <219235043+elvischenv@users.noreply.github.com>
This commit is contained in:
elvischenv 2025-07-24 18:25:41 +08:00 committed by GitHub
parent 2ded067fd2
commit 5a19a6c670
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 6 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
docker/Dockerfile
View File

@ -276,10 +276,6 @@ ARG PYTORCH_CUDA_INDEX_BASE_URL
ENV UV_HTTP_TIMEOUT=500 ENV UV_HTTP_TIMEOUT=500
ENV UV_INDEX_STRATEGY="unsafe-best-match" 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/lint.txt requirements/lint.txt
COPY requirements/test.txt requirements/test.txt COPY requirements/test.txt requirements/test.txt
COPY requirements/dev.txt requirements/dev.txt COPY requirements/dev.txt requirements/dev.txt
@ -452,10 +448,6 @@ ARG PIP_EXTRA_INDEX_URL UV_EXTRA_INDEX_URL
ENV UV_HTTP_TIMEOUT=500 ENV UV_HTTP_TIMEOUT=500
ENV UV_INDEX_STRATEGY="unsafe-best-match" 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) # 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%%.*}"; \ CUDA_MAJOR="${CUDA_VERSION%%.*}"; \

2
docs/contributing/ci/update_pytorch_version.md
View File

@ -134,7 +134,7 @@ MAX_JOBS=16 uv pip install --system \
```bash ```bash
uv pip install --system \ uv pip install --system \
--no-build-isolation "git+https://github.com/state-spaces/mamba@v2.2.4" --no-build-isolation "git+https://github.com/state-spaces/mamba@v2.2.5"
``` ```
### causal-conv1d ### causal-conv1d

2
requirements/test.in
View File

@ -26,7 +26,7 @@ torch==2.7.1
torchaudio==2.7.1 torchaudio==2.7.1
torchvision==0.22.1 torchvision==0.22.1
transformers_stream_generator # required for qwen-vl test transformers_stream_generator # required for qwen-vl test
mamba_ssm # required for plamo2 test mamba_ssm==2.2.5 # required for plamo2 test
matplotlib # required for qwen-vl test matplotlib # required for qwen-vl test
mistral_common[image,audio] >= 1.8.2 # required for voxtral test mistral_common[image,audio] >= 1.8.2 # required for voxtral test
num2words # required for smolvlm test num2words # required for smolvlm test

6
requirements/test.txt
View File

@ -421,7 +421,7 @@ lxml==5.3.0
# sacrebleu # sacrebleu
mako==1.3.10 mako==1.3.10
# via alembic # via alembic
mamba-ssm==2.2.4 mamba-ssm==2.2.5
# via -r requirements/test.in # via -r requirements/test.in
markdown==3.8.2 markdown==3.8.2
# via mlflow # via mlflow
@ -1152,7 +1152,9 @@ transformers==4.53.2
transformers-stream-generator==0.0.5 transformers-stream-generator==0.0.5
# via -r requirements/test.in # via -r requirements/test.in
triton==3.3.1 triton==3.3.1
# via torch # via
# mamba-ssm
# torch
tritonclient==2.51.0 tritonclient==2.51.0
# via # via
# -r requirements/test.in # -r requirements/test.in