Skip to content

Commit 6fce98a

Browse files
authored
fix: Add common and test dependencies to sglang runtime build (#2279) (#2322)
1 parent c92c1f4 commit 6fce98a

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

container/Dockerfile.sglang

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -453,6 +453,16 @@ COPY --from=base /workspace/wheels/nixl/*.whl wheelhouse/
453453
RUN uv pip install "flashinfer-python==0.2.9rc2" --prerelease=allow && \
454454
uv pip install "ai-dynamo[sglang]" --find-links wheelhouse
455455

456+
# Common dependencies
457+
# TODO: Remove extra install and use pyproject.toml to define all dependencies
458+
RUN --mount=type=bind,source=./container/deps/requirements.txt,target=/tmp/requirements.txt \
459+
uv pip install --requirement /tmp/requirements.txt
460+
461+
# Install test dependencies
462+
# TODO: Remove this once we have a functional CI image built on top of the runtime image
463+
RUN --mount=type=bind,source=./container/deps/requirements.test.txt,target=/tmp/requirements.txt \
464+
uv pip install --requirement /tmp/requirements.txt
465+
456466
# Copy launch banner
457467
RUN --mount=type=bind,source=./container/launch_message.txt,target=/workspace/launch_message.txt \
458468
sed '/^#\s/d' /workspace/launch_message.txt > ~/.launch_screen && \

0 commit comments

Comments
 (0)