Compare commits
1 Commits
openai-roc
...
mla-multi-
| Author | SHA1 | Date | |
|---|---|---|---|
|
e52c6f0a98
|
36
Dockerfile
36
Dockerfile
@@ -1,3 +1,35 @@
|
||||
FROM vllm/vllm-openai-rocm:nightly
|
||||
#FROM vllm/vllm-openai:v0.19.0-cu130
|
||||
FROM vllm/vllm-openai:cu130-nightly-x86_64
|
||||
|
||||
RUN pip install av soundfile
|
||||
# Fix the broken ass nightly build that forgot to include pandas
|
||||
RUN pip install --no-cache-dir pandas
|
||||
|
||||
# Install LMCache for KV cache offloading / sharing across nodes
|
||||
# Build with system CUDA 13.0 for Blackwell (B200)
|
||||
RUN apt-get update && apt-get install -y git \
|
||||
libcusolver-dev-13-0 \
|
||||
libcusparse-dev-13-0 \
|
||||
libcublas-dev-13-0 \
|
||||
libcurand-dev-13-0 \
|
||||
libcufft-dev-13-0 \
|
||||
libnvjitlink-dev-13-0 && \
|
||||
git clone https://github.com/Byteflux/LMCache.git /tmp/lmcache && \
|
||||
cd /tmp/lmcache && \
|
||||
git checkout mla-multi-group-kv-cache-with-redis && \
|
||||
CUDA_HOME=/usr/local/cuda \
|
||||
TORCH_CUDA_ARCH_LIST="10.0" \
|
||||
pip install --no-cache-dir --no-build-isolation . && \
|
||||
rm -rf /tmp/lmcache && export CACHE_BUSTER=1
|
||||
|
||||
# Copy over nemotron reasonong parser
|
||||
COPY ./super_v3_reasoning_parser.py /opt/super_v3_reasoning_parser.py
|
||||
|
||||
# Copy over deepseek tool call parser with MTP fixes
|
||||
COPY deepseekv32_tool_parser.py /usr/local/lib/python3.12/dist-packages/vllm/tool_parsers/deepseekv32_tool_parser.py
|
||||
|
||||
# Copy over minimax tool call parser with kwargs fixes
|
||||
COPY minimax_tool_parser.py /usr/local/lib/python3.12/dist-packages/vllm/tool_parsers/minimax_tool_parser.py
|
||||
|
||||
# Copy over minimax parsers with kwargs fixes
|
||||
COPY minimax_tool_parser.py /usr/local/lib/python3.12/dist-packages/vllm/tool_parsers/minimax_tool_parser.py
|
||||
COPY minimax_m2_parser.py /usr/local/lib/python3.12/dist-packages/vllm/parser/minimax_m2_parser.py
|
||||
|
||||
Reference in New Issue
Block a user