| funasr/runtime/onnxruntime/src/punc-model.cpp | ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史 |
funasr/runtime/onnxruntime/src/punc-model.cpp
@@ -1,11 +1,17 @@ #include "precomp.h" namespace funasr { PuncModel *CreatePuncModel(std::map<std::string, std::string>& model_path, int thread_num) PuncModel *CreatePuncModel(std::map<std::string, std::string>& model_path, int thread_num, PUNC_TYPE type) { PuncModel *mm; if (type==PUNC_OFFLINE){ mm = new CTTransformer(); }else if(type==PUNC_ONLINE){ mm = new CTTransformerOnline(); }else{ LOG(ERROR) << "Wrong PUNC TYPE"; exit(-1); } string punc_model_path; string punc_config_path;