Merge pull request #32 from joseporiolayats/whisper-large-v3

adding compatibility with whisper large-v3
absadiki · Feb 9, 2024 · 00eaadf · 00eaadf
2 parents 1aa7982 + 7064550
commit 00eaadf
Showing 1 changed file with 24 additions and 11 deletions.
diff --git a/pywhispercpp/constants.py b/pywhispercpp/constants.py
@@ -25,17 +25,30 @@
 
 MODELS_DIR = Path(user_data_dir(PACKAGE_NAME)) / 'models'
 
-AVAILABLE_MODELS = ["tiny.en",
-                    "tiny",
-                    "base.en",
-                    "base",
-                    "small.en",
-                    "small",
-                    "medium.en",
-                    "medium",
-                    "large-v1",
-                    "large"]
-
+AVAILABLE_MODELS = [
+                "base",
+                "base-q5_1",
+                "base.en",
+                "base.en-q5_1",
+                "large-v1",
+                "large-v2",
+                "large-v2-q5_0",
+                "large-v3",
+                "large-v3-q5_0",
+                "medium",
+                "medium-q5_0",
+                "medium.en",
+                "medium.en-q5_0",
+                "small",
+                "small-q5_1",
+                "small.en",
+                "small.en-q5_1",
+                "tiny",
+                "tiny-q5_1",
+                "tiny.en",
+                "tiny.en-q5_1",
+                "tiny.en-q8_0",
+                ]
 PARAMS_SCHEMA = {  # as exactly presented in whisper.cpp
     'n_threads': {
             'type': int,