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)