Fix initial tokens to decoding (#246)

2023-08-09 12:33:47 +08:00
parent aeb112dd06
commit aa48b76d4b
5 changed files with 10 additions and 5 deletions
--- a/sherpa-onnx/csrc/offline-transducer-greedy-search-decoder.cc
+++ b/sherpa-onnx/csrc/offline-transducer-greedy-search-decoder.cc
@@ -30,8 +30,9 @@ OfflineTransducerGreedySearchDecoder::Decode(Ort::Value encoder_out,

  std::vector<OfflineTransducerDecoderResult> ans(batch_size);
  for (auto &r : ans) {
+    r.tokens.resize(context_size, -1);
    // 0 is the ID of the blank token
-    r.tokens.resize(context_size, 0);
+    r.tokens.back() = 0;
  }

  auto decoder_input = model_->BuildDecoderInput(ans, ans.size());