Dockerfile.b12x-vision 385 B

1234567891011
  1. FROM eugr/spark-vllm-b12x:latest
  2. COPY plugin /opt/dsv4_vision_src
  3. COPY patch_dspark_skip_bias_vl.py /tmp/patch_dspark_skip_bias_vl.py
  4. RUN python3 -m pip install --no-cache-dir --no-deps /opt/dsv4_vision_src \
  5. && python3 /tmp/patch_dspark_skip_bias_vl.py
  6. ENV VLLM_PLUGINS=dsv4_vision
  7. ENV TORCH_CUDA_ARCH_LIST=12.1a
  8. ENV FLASHINFER_CUDA_ARCH_LIST=12.1a
  9. ENTRYPOINT ["vllm", "serve"]