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

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

View File

@@ -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)