From 3d0e14d3a739d633c8188e0cc8a613cfb21a0c6d Mon Sep 17 00:00:00 2001 From: Jian Date: Sat, 11 Jan 2025 15:05:16 -0500 Subject: [PATCH] fixed `get_tokenizer()` -fixed `get_tokenizer()` missing positional argument for `get_valid_language()` --- stable_whisper/whisper_compatibility.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/stable_whisper/whisper_compatibility.py b/stable_whisper/whisper_compatibility.py index cffa672..37115d6 100644 --- a/stable_whisper/whisper_compatibility.py +++ b/stable_whisper/whisper_compatibility.py @@ -328,5 +328,5 @@ def get_tokenizer(model=None, is_faster_model: bool = False, **kwargs): (model.num_languages if hasattr(model, 'num_languages') else model.model.num_languages) elif 'num_languages' in kwargs: del kwargs['num_languages'] - kwargs['language'] = get_valid_language(kwargs.get('language')) + kwargs['language'] = get_valid_language(kwargs.get('language'), is_faster_model, model) return tokenizer(**kwargs)