2025-12-04 23:23:22 +08:00
|
|
|
ARG VLLM_BASE_IMAGE=vllm/vllm-openai
|
2026-02-26 14:36:03 +08:00
|
|
|
ARG VLLM_BASE_TAG=v0.16.0
|
2025-12-04 23:23:22 +08:00
|
|
|
FROM ${VLLM_BASE_IMAGE}:${VLLM_BASE_TAG}
|
|
|
|
|
ARG APP_DIR=/workspace/vllm-omni
|
|
|
|
|
WORKDIR ${APP_DIR}
|
2026-02-03 13:43:45 +08:00
|
|
|
|
2025-12-04 23:23:22 +08:00
|
|
|
COPY . .
|
2026-02-03 13:43:45 +08:00
|
|
|
|
2026-01-16 21:16:43 +08:00
|
|
|
# Install system dependencies
|
|
|
|
|
RUN apt-get update && \
|
2026-02-27 14:34:27 +08:00
|
|
|
apt-get install -y ffmpeg git sox libsox-fmt-all jq && \
|
2026-02-03 13:43:45 +08:00
|
|
|
apt-get clean && \
|
|
|
|
|
rm -rf /var/lib/apt/lists/*
|
2026-01-16 21:16:43 +08:00
|
|
|
|
2026-02-05 10:52:28 +00:00
|
|
|
# Install vLLM wheel for the specific commit
|
|
|
|
|
# URL format: https://wheels.vllm.ai/<commit_hash>/
|
2026-02-26 08:19:53 +00:00
|
|
|
# Current vLLM wheel commit: 6042e66cd5304fc043d96aaa0c22d56f939af320
|
2026-02-09 10:01:39 +00:00
|
|
|
# Note: --prerelease=allow is required because nightly/commit wheels use
|
|
|
|
|
# pre-release version numbers (e.g. 2.10.0.dev...) which uv ignores by default,
|
|
|
|
|
# causing it to fall back to the older stable release from PyPI.
|
2026-02-26 08:19:53 +00:00
|
|
|
RUN uv pip install --system --upgrade --force-reinstall --prerelease=allow vllm --extra-index-url https://wheels.vllm.ai/6042e66cd5304fc043d96aaa0c22d56f939af320
|
2026-02-13 06:39:28 +00:00
|
|
|
|
2026-02-11 22:34:05 +08:00
|
|
|
RUN uv pip uninstall --system nvidia-cublas-cu12
|
2025-12-04 23:23:22 +08:00
|
|
|
# Install vllm-omni into the same uv-managed Python environment used by the base image.
|
2026-02-11 21:31:09 +08:00
|
|
|
# Use bash -c so that $(python3 -c ...) is expanded inside the container.
|
2026-02-13 06:55:10 +00:00
|
|
|
RUN uv pip install --system --no-cache-dir ".[dev]"
|
2026-02-12 09:52:51 +00:00
|
|
|
|
2025-12-04 23:23:22 +08:00
|
|
|
RUN ln -sf /usr/bin/python3 /usr/bin/python
|
2026-02-03 13:43:45 +08:00
|
|
|
|
2025-12-04 23:23:22 +08:00
|
|
|
ENTRYPOINT []
|