CTranslate2 的 Whisper base.en 模型

此存储库包含将 openai/whisper-base.en 转换为 CTranslate2 模型格式的内容。

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

示例

from faster_whisper import WhisperModel

model = WhisperModel("base.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-base.en --output_dir faster-whisper-base.en \
    --copy_files tokenizer.json --quantization float16

请注意模型权重以 FP16 格式保存。在加载模型时可以更改此类型，使用 compute_type option in CTranslate2 。

更多信息

有关原始模型的更多信息，请参阅其 model card 。