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-modified-beam-search-decoder.cc
+++ b/sherpa-onnx/csrc/offline-transducer-modified-beam-search-decoder.cc
@@ -32,7 +32,8 @@ OfflineTransducerModifiedBeamSearchDecoder::Decode(
  int32_t vocab_size = model_->VocabSize();
  int32_t context_size = model_->ContextSize();

-  std::vector<int64_t> blanks(context_size, 0);
+  std::vector<int64_t> blanks(context_size, -1);
+  blanks.back() = 0;

  std::deque<Hypotheses> finalized;
  std::vector<Hypotheses> cur;