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

Minor fixes for rknn (#1925)

Browse Source
This commit is contained in:
Fangjun Kuang
2025-02-26 16:26:18 +08:00
committed by GitHub
parent 2f9a2b20a1
commit 82cb8a5dc3
7 changed files with 46 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
sherpa-onnx/csrc/online-model-config.cc
View File

@@ -7,6 +7,7 @@
#include "sherpa-onnx/csrc/file-utils.h"
#include "sherpa-onnx/csrc/macros.h"
#include "sherpa-onnx/csrc/text-utils.h"
namespace sherpa_onnx {
@@ -65,6 +66,29 @@ bool OnlineModelConfig::Validate() const {
SHERPA_ONNX_LOGE("num_threads should be > 0. Given %d", num_threads);
return false;
}
if (!transducer.encoder.empty() && (EndsWith(transducer.encoder, ".rknn") ||
EndsWith(transducer.decoder, ".rknn") ||
EndsWith(transducer.joiner, ".rknn"))) {
SHERPA_ONNX_LOGE(
"--provider is %s, which is not rknn, but you pass rknn model "
"filenames. encoder: '%s', decoder: '%s', joiner: '%s'",
provider_config.provider.c_str(), transducer.encoder.c_str(),
transducer.decoder.c_str(), transducer.joiner.c_str());
return false;
}
}
if (provider_config.provider == "rknn") {
if (!transducer.encoder.empty() && (EndsWith(transducer.encoder, ".onnx") ||
EndsWith(transducer.decoder, ".onnx") ||
EndsWith(transducer.joiner, ".onnx"))) {
SHERPA_ONNX_LOGE(
"--provider is rknn, but you pass onnx model "
"filenames. encoder: '%s', decoder: '%s', joiner: %'s'",
transducer.encoder.c_str(), transducer.decoder.c_str(),
transducer.joiner.c_str());
return false;
}
}
if (!tokens_buf.empty() && FileExists(tokens)) {