LinTO-STT-Whisper

镜像概述和主要用途

LinTO-STT-Whisper 是一个基于 Whisper 模型 的自动语音识别 (ASR) API。该服务既可以作为独立的转录服务使用，也可以通过消息代理连接器部署在微服务架构中。支持离线或实时转录功能，您可以通过 LinTO Studio 在浏览器中直接试用 LinTO-STT。

核心功能和特性

• 支持多种部署模式：HTTP API、消息队列任务处理和WebSocket流式传输
• 基于Whisper模型的高准确度语音识别
• 支持多语言识别和自动语言检测
• 可配置的语音活动检测(VAD)
• 支持GPU加速以提高性能
• 灵活的模型选择，包括不同大小的Whisper模型和HuggingFace上的模型

使用场景和适用范围

• 离线语音转录服务
• 实时语音转文字应用
• 微服务架构中的语音识别组件
• 需要高准确度语音识别的企业级应用
• 多语言语音处理系统

前置条件

软件要求

• Docker 环境
• 如需GPU加速，需安装 nvidia-container-toolkit

硬件要求

• 至少8GB磁盘空间用于构建Docker镜像，模型文件可能占用高达5GB空间
• 根据模型大小，最高需要7GB内存
• 每个工作进程需要1个CPU核心，推理时间与CPU性能成正比

在GPU上，不同模型大小的显存峰值使用情况如下表所示（根据后端和精度不同）：

模型要求

LinTO-STT-Whisper需要Whisper模型来执行语音识别。如果尚未下载，模型将在首次转录时下载，可能会占用数GB的磁盘空间。

可选的对齐模型（已弃用）

LinTO-STT-Whisper还可以选择使用wav2vec模型执行单词对齐。wav2vec模型可以通过以下方式指定：

• (TorchAudio) 使用对应torchaudio管道的字符串（例如WAV2VEC2_ASR_BASE_960H）
• (HuggingFace's Transformers) 使用对应HuggingFace仓库的字符串（例如jonatasgrosman/wav2vec2-large-xlsr-53-english）
• (SpeechBrain) 使用包含SpeechBrain模型的文件夹路径

默认提供了法语(fr)、英语(en)、西班牙语(es)、德语(de)、荷兰语(nl)、日语(ja)和中文(zh)的wav2vec模型。但建议不再使用对齐模型，因为它已不再需要和测试。

微服务的消息代理和共享文件夹

任务模式下，STT的唯一入口是发布到消息代理的任务。支持的消息代理包括RabbitMQ、Redis和Amazon SQS。此外，为防止大型音频通过消息代理传输，STT-Worker使用共享存储文件夹(SHARED_FOLDER)。

部署指南

1. 构建或拉取镜像

bash
git clone [***]
cd linto-stt
docker build . -f whisper/Dockerfile.ctranslate2 -t linto-stt-whisper:latest

或直接拉取预构建镜像：

bash
docker pull lintoai/linto-stt-whisper

2. 配置.env文件

whisper/.envdefault 提供了.env文件的示例。

MODEL环境变量

警告： 模型将在首次转录时（如有需要将下载并）加载到内存中。当使用来自Hugging Face(transformers)的Whisper模型以及ctranslate2(faster_whisper)时，还将下载torch库以进行从torch到ctranslate2的转换。

如果要预加载模型（稍后指定路径<ASR_PATH>作为MODEL），您可能需要下载以下OpenAI Whisper模型之一：

• 多语言Whisper模型可通过以下链接下载：

  • tiny
  • base
  • small
  • medium
  • large-v1
  • large-v2
  • large-v3
  • large-v3-turbo

• 也可以找到专门用于英语的Whisper模型：

  • tiny.en
  • base.en
  • small.en
  • medium.en

如果您过去曾在本地使用OpenAI-Whipser，模型可以在~/.cache/whisper下找到。

对于来自Hugging Face(transformers)的Whisper模型也是如此，例如[***] Face标识符distil-whisper/distil-large-v2）。

LANGUAGE环境变量

LANGUAGE环境变量可用于设置默认语言（可以是"*"表示自动语言检测）。请注意，language也可以作为请求中的参数传递：在这种情况下，它将覆盖LANGUAGE环境变量。

语言值可以是：

• 通配符"*"，表示自动语言检测（由Whisper模型执行）
• 语言BCP-47代码（"fr-FR"、"en-US"、"yue-HK"等）
• 两或三个字母的语言代码（"fr"、"en"、"yue"等）。请注意，这是BCP-47代码中实际使用的唯一部分。
• 语言名称（"French"、"English"、"Cantonese"等）。

Whisper支持的语言列表： af(南非荷兰语), am(阿姆哈拉语), ar(阿拉伯语), as(阿萨姆语), az(阿塞拜疆语), ba(巴什基尔语), be(白俄罗斯语), bg(保加利亚语), bn(孟加拉语), bo(藏语), br(布列塔尼语), bs(波斯尼亚语), ca(加泰罗尼亚语), cs(捷克语), cy(威尔士语), da(丹麦语), de(德语), el(希腊语), en(英语), es(西班牙语), et(爱沙尼亚语), eu(巴斯克语), fa(波斯语), fi(芬兰语), fo(法罗语), fr(法语), gl(加利西亚语), gu(古吉拉特语), ha(豪萨语), haw(夏威夷语), he(希伯来语), hi(印地语), hr(克罗地亚语), ht(海地克里奥尔语), hu(匈牙利语), hy(亚美尼亚语), id(印度尼西亚语), is(冰岛语), it(意大利语), ja(日语), jw(爪哇语), ka(格鲁吉亚语), kk(哈萨克语), km(高棉语), kn(卡纳达语), ko(韩语), la(拉丁语), lb(卢森堡语), ln(林加拉语), lo(老挝语), lt(立陶宛语), lv(拉脱维亚语), mg(马尔加什语), mi(毛利语), mk(马其顿语), ml(马拉雅拉姆语), mn(蒙古语), mr(马拉地语), ms(马来语), mt(马耳他语), my(缅甸语), ne(尼泊尔语), nl(荷兰语), nn(新挪威语), no(挪威语), oc(奥克语), pa(旁遮普语), pl(波兰语), ps(普什图语