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`." )