콘텐츠로 이동

RBLN NPU에서 whisper-large-v3 서빙

호스트에 RBLN Driver가 설치되어 있어야 하며, RBLN Compiler(rebel-compiler)를 설치하려면 RBLN Portal 계정이 필요합니다. 호환되는 패키지 버전은 설치 매트릭스에서 확인할 수 있습니다.

1
2
3
$ pip install \
  --extra-index-url https://pypi.rbln.ai/simple \
  rebel-compiler==0.12.0

음성 인식에는 audio extra가 필요합니다. 전사 엔드포인트가 업로드된 파일을 읽을 때 쓰는 오디오 디코더를 함께 설치합니다. 이 의존성이 없으면 서버는 시작되지만 첫 음성 전사 요청이 실패합니다.

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"
가이드 — 개요 및 설정

개요

openai/whisper-large-v3는 음성 인식 모델입니다. RBLN NPU에서는 RBLN vLLM 플러그인이 최초 실행 시 컴파일하고, vLLM의 /v1/audio/transcriptions 엔드포인트로 서빙합니다.

Configuration

--rbln-num-devices-per-local-rank는 엔진 프로세스 하나가 사용할 NPU 디바이스 수를 지정하며, RBLN 컴파일러는 Rebellions Scalable Design(RSD)으로 이 디바이스들에 맞게 모델을 분할합니다. NPU별 RSD 적용 방식은 모델 병렬화에서 확인할 수 있습니다. 비전 인코더는 --additional-config(optimum_overrides.visual)로 설정합니다. 이 설정을 바꾸면 다음 실행 시 모델을 재컴파일합니다. 재컴파일을 일으키는 설정 전체는 Compile Cache에 정리되어 있습니다.

이 모델 라인의 다른 크기이며, 각각 별도의 recipe가 있습니다:

whisper-tiny · whisper-base · whisper-small · whisper-medium · whisper-large-v3-turbo

전체 엔진 인자는 설정 가이드에서 확인할 수 있습니다.

참고 및 제약

  • 컴파일된 아티팩트는 최초 실행 시 캐시됩니다. NPU 개수나 엔진 인자를 바꾸면 다음 실행 시 재컴파일됩니다.