Add large-v1 model
This commit is contained in:
@@ -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").
|
||||
|
||||
@@ -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.
|
||||
|
||||
Reference in New Issue
Block a user