Fix broken prompt_reset_on_temperature (#604)

* Fix broken prompt_reset_on_temperature Fixing: SYSTRAN/faster-whisper#603 Broken because `generate_with_fallback()` doesn't return final temperature. Regression since PR356 -> SYSTRAN/faster-whisper#356
AIXerum · Dec 13, 2023 · 57cbea9 · 57cbea9
1 parent 8a58422
commit 57cbea9
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