rnnlm model inference supports num_threads setting (#169)

Co-authored-by: cuidongcai1035 <cuidongcai1035@wezhuiyi.com>
2023-06-07 09:32:27 +08:00
parent 8fad17c87e
commit 9c017c2ccb
10 changed files with 25 additions and 20 deletions
--- a/sherpa-onnx/csrc/online-rnn-lm.h
+++ b/sherpa-onnx/csrc/online-rnn-lm.h
@@ -20,7 +20,7 @@ class OnlineRnnLM : public OnlineLM {
 public:
  ~OnlineRnnLM() override;

-  explicit OnlineRnnLM(const OnlineLMConfig &config);
+  explicit OnlineRnnLM(const OnlineRecognizerConfig &config);

  std::pair<Ort::Value, std::vector<Ort::Value>> GetInitStates() override;