From bb699c3f75030781bca0bd96c82fb5dc0ef82b44 Mon Sep 17 00:00:00 2001 From: Your Name Date: Tue, 16 Jul 2024 15:28:08 -0400 Subject: [PATCH] Add location_trusted to remote diarization auth --- src/wordcab_transcribe/config.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/wordcab_transcribe/config.py b/src/wordcab_transcribe/config.py index 39db791..57abd3a 100644 --- a/src/wordcab_transcribe/config.py +++ b/src/wordcab_transcribe/config.py @@ -109,6 +109,7 @@ def whisper_model_compatibility_check(cls, value: str): # noqa: B902, N805 "large-v1", "large-v2", "large-v3", + "distil-medium.en", "distil-large-v2", "distil-large-v3", ] @@ -117,7 +118,7 @@ def whisper_model_compatibility_check(cls, value: str): # noqa: B902, N805 raise ValueError( "The whisper models must be one of `tiny`, `tiny.en`, `base`," " `base.en`, `small`, `small.en`, `medium`, `medium.en`, `large`," - " `large-v1`, `large-v2`, `large-v3`, `distil-large-v2`, or" + " `large-v1`, `large-v2`, `large-v3`, `distil-medium.en`, `distil-large-v2`, or" " `distil-large-v3`." )