Skip to content

bge-large-en-v1.5 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-large-en-v1.5 is a text-embedding model. On RBLN NPUs, the RBLN vLLM plugin compiles it on first run, then serves it through vLLM's /v1/embeddings 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-small-en-v1.5 · bge-base-en-v1.5

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.