Skip to content
Snippets Groups Projects
Commit 55ba279a authored by Eustache Le Bihan's avatar Eustache Le Bihan
Browse files

fix whisper trfms handle auto language

parent c0eeceb9
No related branches found
No related tags found
No related merge requests found
......@@ -83,6 +83,6 @@ class LightningWhisperSTTHandler(BaseHandler):
logger.debug(f"Language Code Whisper: {language_code}")
if self.start_language == "auto":
language_code += "-auto"
language_code += "-auto"
yield (pred_text, language_code)
......@@ -40,9 +40,8 @@ class WhisperSTTHandler(BaseHandler):
self.torch_dtype = getattr(torch, torch_dtype)
self.compile_mode = compile_mode
self.gen_kwargs = gen_kwargs
if language == 'auto':
language = None
self.last_language = language
self.start_language = language
self.last_language = language if language != "auto" else None
if self.last_language is not None:
self.gen_kwargs["language"] = self.last_language
......@@ -137,7 +136,7 @@ class WhisperSTTHandler(BaseHandler):
console.print(f"[yellow]USER: {pred_text}")
logger.debug(f"Language Code Whisper: {language_code}")
if self.language is None:
if self.start_language == "auto":
language_code += "-auto"
yield (pred_text, language_code)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment