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

Support paraformer. (#95)

Browse Source
This commit is contained in:
Fangjun Kuang
2023-03-28 17:59:54 +08:00
committed by GitHub
parent dffb0fd43c
commit 423d89e9a5
29 changed files with 858 additions and 138 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
sherpa-onnx/csrc/offline-transducer-model.cc
View File

@@ -16,7 +16,7 @@ namespace sherpa_onnx {
class OfflineTransducerModel::Impl {
public:
explicit Impl(const OfflineTransducerModelConfig &config)
explicit Impl(const OfflineModelConfig &config)
: config_(config),
env_(ORT_LOGGING_LEVEL_WARNING),
sess_opts_{},
@@ -24,17 +24,17 @@ class OfflineTransducerModel::Impl {
sess_opts_.SetIntraOpNumThreads(config.num_threads);
sess_opts_.SetInterOpNumThreads(config.num_threads);
{
auto buf = ReadFile(config.encoder_filename);
auto buf = ReadFile(config.transducer.encoder_filename);
InitEncoder(buf.data(), buf.size());
}
{
auto buf = ReadFile(config.decoder_filename);
auto buf = ReadFile(config.transducer.decoder_filename);
InitDecoder(buf.data(), buf.size());
}
{
auto buf = ReadFile(config.joiner_filename);
auto buf = ReadFile(config.transducer.joiner_filename);
InitJoiner(buf.data(), buf.size());
}
}
@@ -164,7 +164,7 @@ class OfflineTransducerModel::Impl {
}
private:
OfflineTransducerModelConfig config_;
OfflineModelConfig config_;
Ort::Env env_;
Ort::SessionOptions sess_opts_;
Ort::AllocatorWithDefaultOptions allocator_;
@@ -195,8 +195,7 @@ class OfflineTransducerModel::Impl {
int32_t context_size_ = 0; // initialized in InitDecoder
};
OfflineTransducerModel::OfflineTransducerModel(
const OfflineTransducerModelConfig &config)
OfflineTransducerModel::OfflineTransducerModel(const OfflineModelConfig &config)
: impl_(std::make_unique<Impl>(config)) {}
OfflineTransducerModel::~OfflineTransducerModel() = default;