From 7064550f7ce90efe61f7942c8a055d1f3e052619 Mon Sep 17 00:00:00 2001
From: Josep Oriol Ayats <oriol.ayats@gmail.com>
Date: Fri, 9 Feb 2024 11:20:30 +0100
Subject: [PATCH] added all the available models

---
 pywhispercpp/constants.py | 37 ++++++++++++++++++++++++-------------
 1 file changed, 24 insertions(+), 13 deletions(-)

diff --git a/pywhispercpp/constants.py b/pywhispercpp/constants.py
index 0ab6bf8..00b0c0a 100644
--- a/pywhispercpp/constants.py
+++ b/pywhispercpp/constants.py
@@ -25,19 +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",
-                    "large-v3",
-                    ]
-
+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,