Add timestamps for offline paraformer (#310)

2023-09-14 19:33:41 +08:00
parent 47184f9db7
commit e2be532b32
11 changed files with 175 additions and 33 deletions
--- a/sherpa-onnx/csrc/offline-paraformer-greedy-search-decoder.h
+++ b/sherpa-onnx/csrc/offline-paraformer-greedy-search-decoder.h
@@ -17,7 +17,8 @@ class OfflineParaformerGreedySearchDecoder : public OfflineParaformerDecoder {
      : eos_id_(eos_id) {}

  std::vector<OfflineParaformerDecoderResult> Decode(
-      Ort::Value log_probs, Ort::Value /*token_num*/) override;
+      Ort::Value log_probs, Ort::Value token_num,
+      Ort::Value us_cif_peak = Ort::Value(nullptr)) override;

 private:
  int32_t eos_id_;