雾聪
2024-03-26 6a8c595fc3ca8005e2d08387354553835c64e951
add log for torch
1个文件已修改
11 ■■■■ 已修改文件
runtime/onnxruntime/src/paraformer-torch.cpp 11 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
runtime/onnxruntime/src/paraformer-torch.cpp
@@ -45,8 +45,15 @@
    #ifdef USE_IPEX
    torch::jit::setTensorExprFuserEnabled(false);
    #endif
    torch::jit::script::Module model = torch::jit::load(am_model, device);
    model_ = std::make_shared<TorchModule>(std::move(model));
    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,