From d48895c34393cae09b375553bcf02d09e3f5b797 Mon Sep 17 00:00:00 2001 From: Dario Pellegrino Date: Fri, 22 Nov 2024 00:16:22 +0100 Subject: [PATCH] large-v3-turbo compatible --- README.md | 2 +- whisper_online.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 9acc83a..1670ba9 100644 --- a/README.md +++ b/README.md @@ -94,7 +94,7 @@ options: --min-chunk-size MIN_CHUNK_SIZE Minimum audio chunk size in seconds. It waits up to this time to do processing. If the processing takes shorter time, it waits, otherwise it processes the whole segment that was received by this time. - --model {tiny.en,tiny,base.en,base,small.en,small,medium.en,medium,large-v1,large-v2,large-v3,large} + --model {tiny.en,tiny,base.en,base,small.en,small,medium.en,medium,large-v1,large-v2,large-v3,large,large-v3-turbo} Name size of the Whisper model to use (default: large-v2). The model is automatically downloaded from the model hub if not present in model cache dir. --model_cache_dir MODEL_CACHE_DIR Overriding the default model cache dir where models downloaded from the hub are saved diff --git a/whisper_online.py b/whisper_online.py index dee2cd8..183b0a0 100644 --- a/whisper_online.py +++ b/whisper_online.py @@ -648,7 +648,7 @@ def add_shared_args(parser): parser: argparse.ArgumentParser object """ parser.add_argument('--min-chunk-size', type=float, default=1.0, help='Minimum audio chunk size in seconds. It waits up to this time to do processing. If the processing takes shorter time, it waits, otherwise it processes the whole segment that was received by this time.') - parser.add_argument('--model', type=str, default='large-v2', choices="tiny.en,tiny,base.en,base,small.en,small,medium.en,medium,large-v1,large-v2,large-v3,large".split(","),help="Name size of the Whisper model to use (default: large-v2). The model is automatically downloaded from the model hub if not present in model cache dir.") + parser.add_argument('--model', type=str, default='large-v2', choices="tiny.en,tiny,base.en,base,small.en,small,medium.en,medium,large-v1,large-v2,large-v3,large,large-v3-turbo".split(","),help="Name size of the Whisper model to use (default: large-v2). The model is automatically downloaded from the model hub if not present in model cache dir.") parser.add_argument('--model_cache_dir', type=str, default=None, help="Overriding the default model cache dir where models downloaded from the hub are saved") parser.add_argument('--model_dir', type=str, default=None, help="Dir where Whisper model.bin and other files are saved. This option overrides --model and --model_cache_dir parameter.") parser.add_argument('--lan', '--language', type=str, default='auto', help="Source language code, e.g. en,de,cs, or 'auto' for language detection.")