Add large-v1 model

This commit is contained in:
Guillaume Klein
2023-03-28 14:36:10 +02:00
parent 8246479fda
commit 0224400584
2 changed files with 3 additions and 2 deletions

View File

@@ -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").