alias large to large-v3

zackees · Jan 7, 2024 · 0d6b37a · 0d6b37a
1 parent 6431c56
commit 0d6b37a
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/transcribe_anything/cmd.py b/transcribe_anything/cmd.py
@@ -23,7 +23,10 @@
     "small.en",
     "medium",
     "medium.en",
+    "large-legacy",
     "large",
+    "large-v2",
+    "large-v3"
 ]
 
 def main() -> int:
@@ -76,6 +79,11 @@ def main() -> int:
     )
     # add extra options that are passed into the transcribe function
     args, unknown = parser.parse_known_args()
+    if args.model == "large-legacy":
+        args.model = "large"
+    elif args.model == "large":
+        args.model = "large-v3"
+
     if unknown:
         print(f"Unknown args: {unknown}")
     print(f"Running transcribe_audio on {args.url_or_file}")