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(