Use log probs for paraformer (#120)

* Use log probs for paraformer * Fix
2023-04-09 23:28:10 +08:00
parent 80060c276d
commit d781fcdeef
3 changed files with 15 additions and 11 deletions
--- a/sherpa-onnx/csrc/offline-paraformer-greedy-search-decoder.h
+++ b/sherpa-onnx/csrc/offline-paraformer-greedy-search-decoder.h
@@ -17,7 +17,7 @@ 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*/) override;

 private:
  int32_t eos_id_;