diff --git a/faster_whisper/transcribe.py b/faster_whisper/transcribe.py index df4fec91..ce7fa999 100644 --- a/faster_whisper/transcribe.py +++ b/faster_whisper/transcribe.py @@ -100,7 +100,8 @@ def __init__( Args: model_size_or_path: Size of the model to use (tiny, tiny.en, base, base.en, - small, small.en, medium, medium.en, large-v1, large-v2, large-v3, or large), a path to a + small, small.en, distil-small.en, medium, medium.en, distil-medium.en, large-v1, + large-v2, large-v3, large, distil-large-v2 or distil-large-v3), a path to a converted model directory, or a CTranslate2-converted Whisper model ID from the HF Hub. When a size or a model ID is configured, the converted model is downloaded from the Hugging Face Hub. diff --git a/faster_whisper/utils.py b/faster_whisper/utils.py index 93ade3a4..481bd748 100644 --- a/faster_whisper/utils.py +++ b/faster_whisper/utils.py @@ -54,8 +54,9 @@ def download_model( Args: size_or_id: Size of the model to download from https://huggingface.co/Systran - (tiny, tiny.en, base, base.en, small, small.en medium, medium.en, large-v1, large-v2, - large-v3, large), or a CTranslate2-converted model ID from the Hugging Face Hub + (tiny, tiny.en, base, base.en, small, small.en, distil-small.en, medium, medium.en, + distil-medium.en, large-v1, large-v2, large-v3, large, distil-large-v2, + distil-large-v3), or a CTranslate2-converted model ID from the Hugging Face Hub (e.g. Systran/faster-whisper-large-v3). output_dir: Directory where the model should be saved. If not set, the model is saved in the cache directory.