Skip to content

Commit 3f7c7a7

Browse files
nv-tusharmatzulingk
authored andcommitted
fix: Add common and test dependencies to sglang runtime build (#2279) (#2322)
1 parent f8b95fd commit 3f7c7a7

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
@@ -454,6 +454,16 @@ COPY --from=base /workspace/wheels/nixl/*.whl wheelhouse/
454454
RUN uv pip install "flashinfer-python==0.2.9rc2" --prerelease=allow && \
455455
uv pip install "ai-dynamo[sglang]" --find-links wheelhouse
456456

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

0 commit comments

Comments
 (0)