From cc22d4db20d5623ae7fde294eefc6ff3df1b31e8 Mon Sep 17 00:00:00 2001
From: Laurent Mazare <laurent.mazare@gmail.com>
Date: Tue, 22 Aug 2023 16:46:34 +0100
Subject: [PATCH] Put the transcribe token before the language one. (#553)

---
 candle-examples/examples/whisper/main.rs | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/candle-examples/examples/whisper/main.rs b/candle-examples/examples/whisper/main.rs
index f0d7cf47..fc64d458 100644
--- a/candle-examples/examples/whisper/main.rs
+++ b/candle-examples/examples/whisper/main.rs
@@ -142,6 +142,9 @@ impl Decoder {
         let mut sum_logprob = 0f64;
         let mut no_speech_prob = f64::NAN;
         let mut tokens = vec![self.sot_token];
+        if let Some(language_token) = self.language_token {
+            tokens.push(language_token);
+        }
         match self.task {
             Some(Task::Transcribe) => tokens.push(self.transcribe_token),
             Some(Task::Translate) => tokens.push(self.translate_token),
@@ -149,9 +152,6 @@ impl Decoder {
                 // Nothing in this case, same as the Python implementation.
             }
         }
-        if let Some(language_token) = self.language_token {
-            tokens.push(language_token);
-        }
         if !self.timestamps {
             tokens.push(self.no_timestamps_token);
         }