语言:

• 英文 标签:
• 音频
• 自动语音识别 许可证: MIT 库名称: ctranslate2

用于CTranslate2的Whisper small.en模型

此仓库包含将openai/whisper-small.en转换为CTranslate2模型格式的结果。

该模型可用于CTranslate2或基于CTranslate2的项目，如faster-whisper。

示例

from faster_whisper import WhisperModel

model = WhisperModel("small.en")

segments, info = model.transcribe("audio.mp3")
for segment in segments:
    print("[%.2fs -> %.2fs] %s" % (segment.start, segment.end, segment.text))

转换详情

原始模型通过以下命令转换：

ct2-transformers-converter --model openai/whisper-small.en --output_dir faster-whisper-small.en \
    --copy_files tokenizer.json --quantization float16

注意模型权重以FP16格式保存。加载模型时可通过CTranslate2中的compute_type选项更改此类型。

更多信息

关于原始模型的更多信息，请参阅其模型卡片。