rnnlm model inference supports num_threads setting (#169)

Co-authored-by: cuidongcai1035 <cuidongcai1035@wezhuiyi.com>
This commit is contained in:
keanu
2023-06-07 09:32:27 +08:00
committed by GitHub
parent 8fad17c87e
commit 9c017c2ccb
10 changed files with 25 additions and 20 deletions

View File

@@ -129,7 +129,7 @@ class OnlineRecognizer::Impl {
endpoint_(config_.endpoint_config) {
if (config.decoding_method == "modified_beam_search") {
if (!config_.lm_config.model.empty()) {
lm_ = OnlineLM::Create(config.lm_config);
lm_ = OnlineLM::Create(config);
}
decoder_ = std::make_unique<OnlineTransducerModifiedBeamSearchDecoder>(