Skip to content

Commit 0015fa9

Browse files
committed
fix: add workaround for installing specific torch and mamba-ssm versions
Signed-off-by: csy1204 <[email protected]>
1 parent 53a77f4 commit 0015fa9

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

docker/Dockerfile

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -162,6 +162,9 @@ ENV UV_HTTP_TIMEOUT=500
162162
COPY requirements/lint.txt requirements/lint.txt
163163
COPY requirements/test.txt requirements/test.txt
164164
COPY requirements/dev.txt requirements/dev.txt
165+
# Workaround for #17068
166+
RUN --mount=type=cache,target=/root/.cache/uv \
167+
uv pip install --system torch==2.6.0 && uv pip install --system mamba-ssm==2.2.4 --no-build-isolation
165168
RUN --mount=type=cache,target=/root/.cache/uv \
166169
uv pip install --system -r requirements/dev.txt
167170
#################### DEV IMAGE ####################
@@ -265,6 +268,9 @@ ADD . /vllm-workspace/
265268
ENV UV_HTTP_TIMEOUT=500
266269

267270
# install development dependencies (for testing)
271+
# Workaround for #17068
272+
RUN --mount=type=cache,target=/root/.cache/uv \
273+
uv pip install --system torch==2.6.0 && uv pip install --system mamba-ssm==2.2.4 --no-build-isolation
268274
RUN --mount=type=cache,target=/root/.cache/uv \
269275
uv pip install --system -r requirements/dev.txt
270276

0 commit comments

Comments
 (0)