Files

36 lines
1.5 KiB
Docker
Raw Permalink Normal View History

2026-04-17 01:45:19 +00:00
#FROM vllm/vllm-openai:v0.19.0-cu130
FROM vllm/vllm-openai:cu130-nightly-x86_64
2026-04-15 10:15:23 +00:00
2026-04-17 02:33:04 +00:00
# Fix the broken ass nightly build that forgot to include pandas
RUN pip install --no-cache-dir pandas
2026-04-15 10:15:23 +00:00
# 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 && \
2026-04-17 01:45:19 +00:00
git clone https://github.com/biondizzle/LMCache.git /tmp/lmcache && \
2026-04-15 10:15:23 +00:00
cd /tmp/lmcache && \
2026-04-17 01:45:19 +00:00
git checkout feat/redis-ttl && \
2026-04-15 10:15:23 +00:00
CUDA_HOME=/usr/local/cuda \
TORCH_CUDA_ARCH_LIST="10.0" \
pip install --no-cache-dir --no-build-isolation . && \
2026-04-17 07:19:15 +00:00
rm -rf /tmp/lmcache && export CACHE_BUSTER=1
2026-04-15 21:05:17 +00:00
2026-04-17 01:45:19 +00:00
# Copy over nemotron reasonong parser
2026-04-15 21:47:35 +00:00
COPY ./super_v3_reasoning_parser.py /opt/super_v3_reasoning_parser.py
2026-04-17 01:45:19 +00:00
# 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
2026-04-17 04:24:36 +00:00
# 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
2026-04-17 04:35:31 +00:00
# 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