From 55ba279a16b1b6a644298edc1c0e879af42bc6d4 Mon Sep 17 00:00:00 2001
From: Eustache Le Bihan <eulebihan@gmail.com>
Date: Tue, 24 Sep 2024 14:42:09 +0200
Subject: [PATCH] fix whisper trfms handle auto language
---
STT/lightning_whisper_mlx_handler.py | 2 +-
STT/whisper_stt_handler.py | 7 +++----
2 files changed, 4 insertions(+), 5 deletions(-)
diff --git a/STT/lightning_whisper_mlx_handler.py b/STT/lightning_whisper_mlx_handler.py
index d0042aa..53b6b5a 100644
--- a/STT/lightning_whisper_mlx_handler.py
+++ b/STT/lightning_whisper_mlx_handler.py
@@ -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)
diff --git a/STT/whisper_stt_handler.py b/STT/whisper_stt_handler.py
index 172a01e..0930087 100644
--- a/STT/whisper_stt_handler.py
+++ b/STT/whisper_stt_handler.py
@@ -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)
--
GitLab