Skip to content

DeepSeek-R1-Distill-Qwen-32B 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

deepseek-ai/DeepSeek-R1-Distill-Qwen-32B is a dense, decoder-only model. On RBLN NPUs, the RBLN vLLM plugin compiles it on first run for a fixed batch size, sequence length, and NPU count, then serves it over vLLM's OpenAI-compatible API.

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:

DeepSeek-R1-Distill-Qwen-1.5B · DeepSeek-R1-Distill-Qwen-7B · DeepSeek-R1-Distill-Qwen-14B

See the Configuration Guide for all engine arguments.

Notes and limitations

  • The batch size, sequence length, and NPU count are fixed at compilation; changing any of these settings triggers recompilation at the next startup.
  • The recipe serves with --max-num-seqs 1. Increase it for higher throughput; the model recompiles to match.