From 1da75ee3c04cfc4b77d3aec7df152ed710128b99 Mon Sep 17 00:00:00 2001 From: xsjk <53306125+xsjk@users.noreply.github.com> Date: Wed, 7 Aug 2024 15:38:34 +0800 Subject: [PATCH] Fix typo in offline-lm-config.cc (#1229) --- sherpa-onnx/python/csrc/offline-lm-config.cc | 2 +- sherpa-onnx/python/sherpa_onnx/offline_recognizer.py | 2 ++ 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/sherpa-onnx/python/csrc/offline-lm-config.cc b/sherpa-onnx/python/csrc/offline-lm-config.cc index ba9ea94d..42edb656 100644 --- a/sherpa-onnx/python/csrc/offline-lm-config.cc +++ b/sherpa-onnx/python/csrc/offline-lm-config.cc @@ -15,7 +15,7 @@ void PybindOfflineLMConfig(py::module *m) { py::class_(*m, "OfflineLMConfig") .def(py::init(), py::arg("model"), py::arg("scale") = 0.5f, - py::arg("lm_num_threads") = 1, py::arg("lm-provider") = "cpu") + py::arg("lm_num_threads") = 1, py::arg("lm_provider") = "cpu") .def_readwrite("model", &PyClass::model) .def_readwrite("scale", &PyClass::scale) .def_readwrite("lm_provider", &PyClass::lm_provider) diff --git a/sherpa-onnx/python/sherpa_onnx/offline_recognizer.py b/sherpa-onnx/python/sherpa_onnx/offline_recognizer.py index dc1fcdf1..e96271a5 100644 --- a/sherpa-onnx/python/sherpa_onnx/offline_recognizer.py +++ b/sherpa-onnx/python/sherpa_onnx/offline_recognizer.py @@ -156,6 +156,8 @@ class OfflineRecognizer(object): lm_config = OfflineLMConfig( model=lm, scale=lm_scale, + lm_num_threads=num_threads, + lm_provider=provider, ) recognizer_config = OfflineRecognizerConfig(