Mirroristas/whisper
2
0
Fork 0
mirror of https://github.com/openai/whisper.git synced 2025-03-30 14:28:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Return the best only if all fallbacks failed

Browse Source
This commit is contained in:
Théo Boyer 2023-05-21 16:58:12 +02:00
parent a72a04414b
commit f677284d11
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
whisper/transcribe.py
View File

@ -148,6 +148,7 @@ def transcribe(
temperatures = (
[temperature] if isinstance(temperature, (int, float)) else temperature
)
decode_result = None
results = {}
for t in temperatures:
@ -183,8 +184,11 @@ def transcribe(
needs_fallback = False # silence
if not needs_fallback:
break
else:
# all failed
return max(results.values(), key=lambda r: r.avg_logprob)
return max(results.values(), key=lambda r: r.avg_logprob)
return decode_result
seek = 0
input_stride = exact_div(