Return the best only if all fallbacks failed

openai · May 21, 2023 · f677284 · f677284
1 parent a72a044
commit f677284
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/whisper/transcribe.py b/whisper/transcribe.py
@@ -148,6 +148,7 @@ def decode_with_fallback(segment: torch.Tensor) -> DecodingResult:
         temperatures = (
             [temperature] if isinstance(temperature, (int, float)) else temperature
         )
+        decode_result = None
         results = {}
 
         for t in temperatures:
@@ -183,8 +184,11 @@ def decode_with_fallback(segment: torch.Tensor) -> DecodingResult:
                 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(