tzhouam
7aea8e8d91
- Introduced OmniMRotaryEmbedding to extend MRotaryEmbedding with methods for computing input positions across various multimodal scenarios, including image, video, and audio inputs. - Updated patch.py to replace the original MRotaryEmbedding with OmniMRotaryEmbedding in the module patching logic. - Created new files for the OmniMRotaryEmbedding implementation and its associated methods. Signed-off-by: [Your Name] <[Your Email]> Signed-off-by: tzhouam <tzhouam@connect.ust.hk>
22 lines
817 B
Docker
22 lines
817 B
Docker
ARG VLLM_BASE_IMAGE=vllm/vllm-openai
|
|
ARG VLLM_BASE_TAG=v0.15.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 && \
|
|
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 commit: f176443446f659dbab5315e056e605d8984fd976 (v0.15.0)
|
|
RUN uv pip install --system --upgrade --force-reinstall vllm --extra-index-url https://wheels.vllm.ai/f176443446f659dbab5315e056e605d8984fd976
|
|
|
|
# Install vllm-omni into the same uv-managed Python environment used by the base image.
|
|
RUN uv pip install --system --no-cache-dir ".[dev]"
|
|
RUN ln -sf /usr/bin/python3 /usr/bin/python
|
|
ENTRYPOINT []
|