Fix a bug for multilingual ASR

k2-fsa · Aug 17, 2023 · 148798b · 148798b
1 parent 54eca59
commit 148798b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -1,7 +1,7 @@
 cmake_minimum_required(VERSION 3.13 FATAL_ERROR)
 project(sherpa-onnx)
 
-set(SHERPA_ONNX_VERSION "1.7.8")
+set(SHERPA_ONNX_VERSION "1.7.9")
 
 # Disable warning about
 #

diff --git a/sherpa-onnx/csrc/offline-whisper-greedy-search-decoder.cc b/sherpa-onnx/csrc/offline-whisper-greedy-search-decoder.cc
@@ -136,8 +136,10 @@ OfflineWhisperGreedySearchDecoder::Decode(Ort::Value cross_k,
   auto logits_shape = logits.GetTensorTypeAndShapeInfo().GetShape();
   int32_t vocab_size = logits_shape[2];
 
-  int32_t max_token_id = static_cast<int32_t>(std::distance(
-      p_logits, std::max_element(p_logits, p_logits + vocab_size)));
+  const float *p_start = p_logits + (logits_shape[1] - 1) * vocab_size;
+
+  int32_t max_token_id = static_cast<int32_t>(
+      std::distance(p_start, std::max_element(p_start, p_start + vocab_size)));
 
   int32_t n_text_ctx = model_->TextCtx();