From 57ca9e910282a98c8c08f790b29de32c1c64bbcc Mon Sep 17 00:00:00 2001 From: Yuta Hayashibe Date: Sat, 24 Sep 2022 13:06:22 +0900 Subject: [PATCH] Add break for fallback --- whisper_streaming/transcriber.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/whisper_streaming/transcriber.py b/whisper_streaming/transcriber.py index 88c4951..8ac4169 100644 --- a/whisper_streaming/transcriber.py +++ b/whisper_streaming/transcriber.py @@ -109,10 +109,10 @@ class WhisperStreamingTranscriber: and result.avg_logprob < self.config.logprob_threshold for result in results ] - logger.debug( - f"Fall back with temperature: {t}, needs_fallback: {needs_fallback}" - ) if any(needs_fallback): + logger.debug( + f"Fall back with temperature: {t}, needs_fallback: {needs_fallback}" + ) _decode_options2: DecodingOptions = self._get_decoding_options( t=t, prompt=self.buffer_tokens, @@ -127,6 +127,8 @@ class WhisperStreamingTranscriber: np.nonzero(needs_fallback)[0] ): results[original_index] = retries[retry_index] + else: + break logger.debug(f"# of results: {len(results)}") return results