vllm-rerank.yml 698 B

123456789101112131415161718192021
  1. vllm-rerank:
  2. image: vllm/vllm-openai-cpu:latest
  3. command: >
  4. --model ${VLLM_RERANK_MODEL:-BAAI/bge-reranker-v2-m3}
  5. --port ${VLLM_RERANK_PORT:-8000}
  6. --dtype float32
  7. --api-key ${VLLM_RERANK_API_KEY}
  8. ${VLLM_RERANK_EXTRA_ARGS:-}
  9. ports:
  10. - "${VLLM_RERANK_PORT:-8000}:${VLLM_RERANK_PORT:-8000}"
  11. volumes:
  12. - vllm_rerank_cache:/root/.cache/huggingface
  13. healthcheck:
  14. test:
  15. - CMD-SHELL
  16. - 'PORT_HEX="$(printf ''%04X'' ${VLLM_RERANK_PORT:-8000})"; cat /proc/net/tcp /proc/net/tcp6 2>/dev/null | grep -q ":$${PORT_HEX} "'
  17. interval: 5s
  18. timeout: 3s
  19. retries: 120
  20. start_period: 10s
  21. restart: unless-stopped