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-lm.cc
+++ b/sherpa-onnx/csrc/offline-lm.cc
@@ -12,7 +12,8 @@

 namespace sherpa_onnx {

-std::unique_ptr<OfflineLM> OfflineLM::Create(const OfflineLMConfig &config) {
+std::unique_ptr<OfflineLM> OfflineLM::Create(
+    const OfflineRecognizerConfig &config) {
  return std::make_unique<OfflineRnnLM>(config);
 }