fixed get_tokenizer()

-fixed `get_tokenizer()` missing positional argument for `get_valid_language()`
jianfch · Jan 11, 2025 · 3d0e14d · 3d0e14d
1 parent cd9232e
commit 3d0e14d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)