Fix broken prompt_reset_on_temperature (SYSTRAN#604)

* Fix broken prompt_reset_on_temperature Fixing: SYSTRAN#603 Broken because `generate_with_fallback()` doesn't return final temperature. Regression since PR356 -> SYSTRAN#356
metame-ai · Jan 2, 2024 · 18e53ad · 18e53ad
1 parent d3ec873
commit 18e53ad
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/faster_whisper/transcribe.py b/faster_whisper/transcribe.py
@@ -737,6 +737,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