Skip to content

whisper-base 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

Speech transcription requires the audio extra, which installs the decoders used to process uploaded audio. Without it, the server starts, but the first transcription request fails.

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

Overview

openai/whisper-base is a speech transcription model. On RBLN NPUs, the RBLN vLLM plugin compiles it on first run, then serves it through vLLM's /v1/audio/transcriptions 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. The vision encoder is configured through --additional-config (optimum_overrides.visual). Changing these settings recompiles the model on the next start. Compile Cache lists every setting that does.

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.