EngineX-Iluvatar/enginex-mr_series-sherpa-onnx
Archived
16
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
xsjk
2024-08-07 15:38:34 +08:00
committed by GitHub
parent 9ee2943ed4
commit 1da75ee3c0
2 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
sherpa-onnx/python/csrc/offline-lm-config.cc
View File

@@ -15,7 +15,7 @@ void PybindOfflineLMConfig(py::module *m) {
py::class_<PyClass>(*m, "OfflineLMConfig") py::class_<PyClass>(*m, "OfflineLMConfig")
.def(py::init<const std::string &, float, int32_t, const std::string &>(), .def(py::init<const std::string &, float, int32_t, const std::string &>(),
py::arg("model"), py::arg("scale") = 0.5f, 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("model", &PyClass::model)
.def_readwrite("scale", &PyClass::scale) .def_readwrite("scale", &PyClass::scale)
.def_readwrite("lm_provider", &PyClass::lm_provider) .def_readwrite("lm_provider", &PyClass::lm_provider)

2
sherpa-onnx/python/sherpa_onnx/offline_recognizer.py
View File

@@ -156,6 +156,8 @@ class OfflineRecognizer(object):
lm_config = OfflineLMConfig( lm_config = OfflineLMConfig(
model=lm, model=lm,
scale=lm_scale, scale=lm_scale,
lm_num_threads=num_threads,
lm_provider=provider,
) )
recognizer_config = OfflineRecognizerConfig( recognizer_config = OfflineRecognizerConfig(