Skip to content

Commit

Permalink
Avoid computing higher temperatures on no_speech segments (openai#1279)
Browse files Browse the repository at this point in the history
* Avoid computing higher temperatures on no_speech

In decode_with_fallback, we compute higher temperatures in the case where compression_ratio is too high or avg_logprob is too low.
But as the computation of no_speech_prob doens't depend on sampling, we can avoid computing higher temperatures if we detect in the first one that the no_speech condition is fulfilled

* Update transcribe.py

---------

Co-authored-by: Jong Wook Kim <jongwook@openai.com>
  • Loading branch information
2 people authored and ilanit1997 committed May 16, 2023
1 parent 07b6909 commit 792f612
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion whisper/transcribe.py
Original file line number Diff line number Diff line change
Expand Up @@ -174,7 +174,11 @@ def decode_with_fallback(segment: torch.Tensor) -> DecodingResult:
and decode_result.avg_logprob < logprob_threshold
):
needs_fallback = True # average log probability is too low

if (
no_speech_threshold is not None
and decode_result.no_speech_prob > no_speech_threshold
):
needs_fallback = False # silence
if not needs_fallback:
break

Expand Down

0 comments on commit 792f612

Please sign in to comment.