github/faster-whisper
1
1
Fork 0
mirror of https://github.com/SYSTRAN/faster-whisper.git synced 2026-01-09 21:48:08 -05:00
Code Issues Packages Projects Releases Wiki Activity

Update WhisperModel documentation to list all available models (#1137)

Browse Source
This commit is contained in:
Mahmoud Ashraf
2024-11-13 18:26:01 +02:00
committed by GitHub
parent c2bf036234
commit 8f01aee36b
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
faster_whisper/transcribe.py
View File

@@ -546,9 +546,9 @@ class WhisperModel:
Args: Args:
model_size_or_path: Size of the model to use (tiny, tiny.en, base, base.en, model_size_or_path: Size of the model to use (tiny, tiny.en, base, base.en,
small, small.en, distil-small.en, medium, medium.en, distil-medium.en, large-v1, 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 large-v2, large-v3, large, distil-large-v2, distil-large-v3, large-v3-turbo, or turbo),
converted model directory, or a CTranslate2-converted Whisper model ID from the HF Hub. a path to a converted model directory, or a CTranslate2-converted Whisper model ID from
When a size or a model ID is configured, the converted model is downloaded the HF Hub. When a size or a model ID is configured, the converted model is downloaded
from the Hugging Face Hub. from the Hugging Face Hub.
device: Device to use for computation ("cpu", "cuda", "auto"). device: Device to use for computation ("cpu", "cuda", "auto").
device_index: Device ID to use. device_index: Device ID to use.