Dockerfile 386 B

123456789101112
  1. ARG BASE_IMAGE
  2. FROM ${BASE_IMAGE}
  3. COPY plugin /opt/dsv4_vision_src
  4. COPY patch_dspark_skip_bias_vl.py /tmp/patch_dspark_skip_bias_vl.py
  5. RUN python3 -m pip install --no-cache-dir --no-deps /opt/dsv4_vision_src \
  6. && python3 /tmp/patch_dspark_skip_bias_vl.py
  7. ENV VLLM_PLUGINS=dsv4_vision
  8. ENV TORCH_CUDA_ARCH_LIST=12.1a
  9. ENV FLASHINFER_CUDA_ARCH_LIST=12.1a
  10. ENTRYPOINT ["vllm", "serve"]