tzhouam
ae818f4b45
The upstream vLLM Qwen2_5_VisionAttention.forward() now requires a `sequence_lengths` argument, but Qwen3_VisionBlock.forward() does not pass it, causing a TypeError at init time. Instead of patching the vLLM package directly, introduce a Qwen3Omni_VisionTransformer subclass in vllm-omni that overrides forward() to properly compute and propagate sequence_lengths via MMEncoderAttention (following qwen3_vl.py's pattern). Both thinker and talker now import the fixed subclass. Also bump the vLLM wheel to e196028dbaa41b. Made-with: Cursor Signed-off-by: tzhouam <tzhouam@connect.ust.hk>
34 lines
1.2 KiB
Docker
34 lines
1.2 KiB
Docker
ARG VLLM_BASE_IMAGE=vllm/vllm-openai
|
|
ARG VLLM_BASE_TAG=v0.16.0
|
|
FROM ${VLLM_BASE_IMAGE}:${VLLM_BASE_TAG}
|
|
ARG APP_DIR=/workspace/vllm-omni
|
|
WORKDIR ${APP_DIR}
|
|
|
|
COPY . .
|
|
|
|
# Install system dependencies
|
|
RUN apt-get update && \
|
|
apt-get install -y ffmpeg git sox libsox-fmt-all jq && \
|
|
apt-get clean && \
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
# Install vLLM wheel for the specific commit
|
|
# URL format: https://wheels.vllm.ai/<commit_hash>/
|
|
# Current vLLM wheel commit: 3dd142adb7f7fa3b0b8345ee046ba36a12d9e530 (releases/v0.17.0)
|
|
# 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.
|
|
RUN uv pip install --system --upgrade --force-reinstall --prerelease=allow vllm --extra-index-url https://wheels.vllm.ai/e196028dbaa41b8e3d1c27b7dbd026403af76c27
|
|
|
|
RUN uv pip install --system --no-cache-dir ".[dev]"
|
|
RUN uv pip install --system --upgrade \
|
|
"flashinfer-cubin==0.6.4" \
|
|
"nvidia-cublas-cu12==12.9.1.4" \
|
|
"numpy==2.2.6"
|
|
RUN uv pip install --system --upgrade \
|
|
"flashinfer-jit-cache==0.6.4" \
|
|
--index-url https://flashinfer.ai/whl/cu129
|
|
RUN ln -sf /usr/bin/python3 /usr/bin/python
|
|
|
|
ENTRYPOINT []
|