Fix broken prompt_reset_on_temperature (#604)

* Fix broken prompt_reset_on_temperature Fixing: #603 Broken because `generate_with_fallback()` doesn't return final temperature. Regression since PR356 -> #356
SYSTRAN · Dec 13, 2023 · ebcfd6b · ebcfd6b
1 parent 19329a3
commit ebcfd6b
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/faster_whisper/transcribe.py b/faster_whisper/transcribe.py
@@ -731,6 +731,13 @@ def generate_with_fallback(
             decode_result = max(
                 below_cr_threshold_results or all_results, key=lambda x: x[1]
             )
+            # to pass final temperature for prompt_reset_on_temperature
+            decode_result = (
+                decode_result[0],
+                decode_result[1],
+                temperature,
+                decode_result[3],
+            )
 
         return decode_result