speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

adapt to new model struct

Browse Source
This commit is contained in:
雾聪 2024-05-27 19:15:58 +08:00
parent 365c820d66
commit 7c11a0d6a8
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
runtime/onnxruntime/src/paraformer-torch.cpp
View File

@ -16,7 +16,7 @@ ParaformerTorch::ParaformerTorch()
}
// offline
void ParaformerTorch::InitAsr(const std::string &am_model, const std::string &am_cmvn, const std::string &am_config, int thread_num){
void ParaformerTorch::InitAsr(const std::string &am_model, const std::string &am_cmvn, const std::string &am_config, const std::string &token_file, int thread_num){
LoadConfigFromYaml(am_config.c_str());
// knf options
fbank_opts_.frame_opts.dither = 0;
@ -28,8 +28,8 @@ void ParaformerTorch::InitAsr(const std::string &am_model, const std::string &am
fbank_opts_.energy_floor = 0;
fbank_opts_.mel_opts.debug_mel = false;
vocab = new Vocab(am_config.c_str());
phone_set_ = new PhoneSet(am_config.c_str());
vocab = new Vocab(token_file.c_str());
phone_set_ = new PhoneSet(token_file.c_str());
LoadCmvn(am_cmvn.c_str());
torch::DeviceType device = at::kCPU;

2
runtime/onnxruntime/src/paraformer-torch.h
View File

@ -42,7 +42,7 @@ namespace funasr {
public:
ParaformerTorch();
~ParaformerTorch();
void InitAsr(const std::string &am_model, const std::string &am_cmvn, const std::string &am_config, int thread_num);
void InitAsr(const std::string &am_model, const std::string &am_cmvn, const std::string &am_config, const std::string &token_file, int thread_num);
void InitHwCompiler(const std::string &hw_model, int thread_num);
void InitSegDict(const std::string &seg_dict_model);
std::vector<std::vector<float>> CompileHotwordEmbedding(std::string &hotwords);