Log a warning when the model is English-only but the language is set to something else (#454)

This commit is contained in:
Guillaume Klein
2023-09-04 11:55:40 +02:00
committed by GitHub
parent 1e6eb967c9
commit 4a41746e55

View File

@@ -315,6 +315,13 @@ class WhisperModel:
language_probability, language_probability,
) )
else: else:
if not self.model.is_multilingual and language != "en":
self.logger.warning(
"The current model is English-only but the language parameter is set to '%s'; "
"using 'en' instead." % language
)
language = "en"
language_probability = 1 language_probability = 1
tokenizer = Tokenizer( tokenizer = Tokenizer(