From f5c0e44935e2861cef8d78293e6206aeae856e4c Mon Sep 17 00:00:00 2001 From: Guillaume Klein Date: Wed, 22 Feb 2023 14:59:29 +0100 Subject: [PATCH] Update README.md --- README.md | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 980b8a9..2c69975 100644 --- a/README.md +++ b/README.md @@ -53,10 +53,12 @@ GPU execution requires the NVIDIA libraries cuBLAS 11.x and cuDNN 8.x to be inst ### Model conversion -A Whisper model should be first converted into the CTranslate2 format. For example the command below converts the "medium" Whisper model and saves the weights in FP16: +A Whisper model should be first converted into the CTranslate2 format. We provide a script to download and convert models from the [Hugging Face model repository](https://huggingface.co/models?sort=downloads&search=whisper). + +For example the command below converts the "large-v2" Whisper model and saves the weights in FP16: ```bash -ct2-transformers-converter --model openai/whisper-medium --output_dir whisper-medium-ct2 --quantization float16 +ct2-transformers-converter --model openai/whisper-large-v2 --output_dir whisper-large-v2-ct2 --quantization float16 ``` If needed, models can also be converted from the code. See the [conversion API](https://opennmt.net/CTranslate2/python/ctranslate2.converters.TransformersConverter.html).