Add support for Whisper models distil-large-v3 and distil-large-v3-tu…

…rbo to Whisper module
Woolverine94 · Oct 3, 2024 · 42bcf08 · 42bcf08
1 parent 51891db
commit 42bcf08
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/ressources/whisper.py b/ressources/whisper.py
@@ -26,11 +26,13 @@
 
 model_list_whisper_builtin = {
     "openai/whisper-tiny": "model.safetensors",
-    "openai/whisper-base": "model.safetensors",        
+    "openai/whisper-base": "model.safetensors",
     "openai/whisper-medium": "model.safetensors",
     "openai/whisper-large": "model.safetensors",
-    "openai/whisper-large-v3": "model.safetensors", 
-    "distil-whisper/distil-large-v2": "model.safetensors", 
+    "openai/whisper-large-v3": "model.safetensors",
+    "distil-whisper/distil-large-v2": "model.safetensors",
+    "distil-whisper/distil-large-v3": "model.safetensors",
+    "openai/whisper-large-v3-turbo": "model.safetensors",
 }
 
 model_list_whisper.update(model_list_whisper_builtin)