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/offline-rnn-lm.h
+++ b/sherpa-onnx/csrc/offline-rnn-lm.h
@@ -17,7 +17,7 @@ class OfflineRnnLM : public OfflineLM {
 public:
  ~OfflineRnnLM() override;

-  explicit OfflineRnnLM(const OfflineLMConfig &config);
+  explicit OfflineRnnLM(const OfflineRecognizerConfig &config);

  /** Rescore a batch of sentences.
   *