support distil-whisper (#557)

2024-01-24 17:17:12 +08:00
parent 72ff979a2e
commit ad3c83045b
4 changed files with 76 additions and 3 deletions
--- a/faster_whisper/utils.py
+++ b/faster_whisper/utils.py
@@ -22,6 +22,9 @@ _MODELS = {
    "large-v2": "Systran/faster-whisper-large-v2",
    "large-v3": "Systran/faster-whisper-large-v3",
    "large": "Systran/faster-whisper-large-v3",
+    "distil-large-v2": "Systran/faster-distil-whisper-large-v2",
+    "distil-medium.en": "Systran/faster-distil-whisper-medium.en",
+    "distil-small.en": "Systran/faster-distil-whisper-small.en",
 }