Add distil models to WhisperModel init and download_model docstrings (#847)

* chore: add distil models to WhisperModel init docstring and download_model docstring
This commit is contained in:
Napuh
2024-05-20 03:51:22 +02:00
committed by GitHub
parent 4acdb5c619
commit f53be1e811
2 changed files with 5 additions and 3 deletions

View File

@@ -100,7 +100,8 @@ 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, 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.