FunASR/runtime/onnxruntime/src/cam-sv-model.cpp
2024-08-01 14:52:07 +08:00

24 lines
738 B
C++

#include "precomp.h"
namespace funasr
{
SvModel *CreateSVModel(std::map<std::string, std::string> &model_path, int thread_num)
{
SvModel *mm;
mm = new CamPPlusSv();
string vad_model_path;
string vad_config_path;
vad_model_path = PathAppend(model_path.at(SV_DIR), MODEL_NAME);
if (model_path.find(SV_QUANT) != model_path.end() && model_path.at(SV_QUANT) == "true")
{
vad_model_path = PathAppend(model_path.at(SV_DIR), QUANT_MODEL_NAME);
}
vad_config_path = PathAppend(model_path.at(SV_DIR), SV_CONFIG_NAME);
mm->InitSv(vad_model_path, vad_config_path, thread_num);
return mm;
}
} // namespace funasr