| runtime/onnxruntime/src/paraformer-torch.cpp | ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史 |
runtime/onnxruntime/src/paraformer-torch.cpp
@@ -45,8 +45,15 @@ #ifdef USE_IPEX torch::jit::setTensorExprFuserEnabled(false); #endif try { torch::jit::script::Module model = torch::jit::load(am_model, device); model_ = std::make_shared<TorchModule>(std::move(model)); LOG(INFO) << "Successfully load model from " << am_model; } catch (std::exception const &e) { LOG(ERROR) << "Error when load am model: " << e.what(); exit(-1); } } void ParaformerTorch::InitLm(const std::string &lm_file,