mirrors/whispering
1
0
Fork 0
mirror of https://github.com/shirayu/whispering.git synced 2024-05-20 09:18:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add break for fallback

Browse source
This commit is contained in:
Yuta Hayashibe 2022-09-24 13:06:22 +09:00
parent 0ad6976d46
commit 57ca9e9102
1 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
whisper_streaming/transcriber.py
View file

@ -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