Skip to content

bge-reranker-large on RBLN NPU

Install the RBLN Driver on the host. An RBLN Portal account is required to install the RBLN Compiler (rebel-compiler). See the installation matrix for compatible package versions.

1
2
3
$ pip install \
  --extra-index-url https://pypi.rbln.ai/simple \
  rebel-compiler==0.12.0
1
2
3
4
$ pip install \
  --extra-index-url https://wheels.vllm.ai/0.30.0/cpu \
  --extra-index-url https://download.pytorch.org/whl/cpu \
  vllm-rbln==0.12.0
Guide — overview and configuration

Overview

BAAI/bge-reranker-large is a reranking model. On RBLN NPUs, the RBLN vLLM plugin compiles it on first run, then serves it through vLLM's /rerank endpoint.

Configuration

--rbln-num-devices-per-local-rank sets how many NPU devices one engine process uses, and the RBLN Compiler partitions the model for those devices with Rebellions Scalable Design (RSD). See Model Parallelism for how RSD applies on each NPU. Changing these settings recompiles the model on the next start. Compile Cache lists every setting that does.

Recipes for other model sizes:

bge-reranker-base

See the Configuration Guide for all engine arguments.

Notes and limitations

  • The compiled artifact is cached on first run; changing the NPU count or engine arguments recompiles on the next start.