From 0c4f2537022536cc87f69c805018d21042db8a88 Mon Sep 17 00:00:00 2001 From: colrobloxkid69420 Date: Tue, 14 Nov 2023 22:38:53 +0200 Subject: [PATCH] added more large models to selection --- transcribe_demo.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/transcribe_demo.py b/transcribe_demo.py index 14ec910..718c910 100644 --- a/transcribe_demo.py +++ b/transcribe_demo.py @@ -17,7 +17,7 @@ def main(): parser = argparse.ArgumentParser() parser.add_argument("--model", default="medium", help="Model to use", - choices=["tiny", "base", "small", "medium", "large"]) + choices=["tiny", "base", "small", "medium", "large", "large-v1", "large-v2", "large-v3"]) parser.add_argument("--non_english", action='store_true', help="Don't use the english model.") parser.add_argument("--energy_threshold", default=1000, @@ -64,7 +64,7 @@ def main(): # Load / Download model model = args.model - if args.model != "large" and not args.non_english: + if args.model not in {"large", "large-v1", "large-v2", "large-v3"} and not args.non_english: model = model + ".en" audio_model = whisper.load_model(model)