[WIP/TEST] distil-wisperの処理を追加

2024-02-04 01:03:38 +09:00
parent 801d948513
commit 7aafce6e2e
6 changed files with 26 additions and 8 deletions
--- a/models/transcription/transcription_whisper.py
+++ b/models/transcription/transcription_whisper.py
@@ -15,6 +15,9 @@ _MODELS = {
    "large-v1": "Systran/faster-whisper-large-v1",
    "large-v2": "Systran/faster-whisper-large-v2",
    "large-v3": "Systran/faster-whisper-large-v3",
+    "distil-small": "Systran/faster-distil-whisper-small.en",
+    "distil-medium": "Systran/faster-distil-whisper-medium.en",
+    "distil-large-v2": "Systran/faster-distil-whisper-large-v2"
 }

 _FILENAMES = [