diff --git a/faster_whisper/transcribe.py b/faster_whisper/transcribe.py index 911928f..96d2e7c 100644 --- a/faster_whisper/transcribe.py +++ b/faster_whisper/transcribe.py @@ -69,7 +69,7 @@ class WhisperModel: Args: model_size_or_path: Size of the model to use (tiny, tiny.en, base, base.en, - small, small.en, medium, medium.en, or large-v2) or a path to a converted + small, small.en, medium, medium.en, large-v1, or large-v2) or a path to a converted model directory. When a size is configured, the converted model is downloaded from the Hugging Face Hub. device: Device to use for computation ("cpu", "cuda", "auto"). diff --git a/faster_whisper/utils.py b/faster_whisper/utils.py index b97c9a1..be61ff9 100644 --- a/faster_whisper/utils.py +++ b/faster_whisper/utils.py @@ -13,6 +13,7 @@ _MODELS = ( "small", "medium.en", "medium", + "large-v1", "large-v2", ) @@ -28,7 +29,7 @@ def download_model( Args: size: Size of the model to download (tiny, tiny.en, base, base.en, small, small.en, - medium, medium.en, or large-v2). + medium, medium.en, large-v1, or large-v2). output_dir: Directory where the model should be saved. If not set, the model is saved in the standard Hugging Face cache directory. show_progress_bars: Show the tqdm progress bars during the download.