Fix typo in offline-lm-config.cc (#1229)

k2-fsa · Aug 7, 2024 · 1da75ee · 1da75ee
1 parent 9ee2943
commit 1da75ee
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git 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_<PyClass>(*m, "OfflineLMConfig")
       .def(py::init<const std::string &, float, int32_t, const std::string &>(),
            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
@@ -156,6 +156,8 @@ def from_transducer(
         lm_config = OfflineLMConfig(
             model=lm,
             scale=lm_scale,
+            lm_num_threads=num_threads,
+            lm_provider=provider,
         )
 
         recognizer_config = OfflineRecognizerConfig(