From 16a3cd3cfb866e99b68417e36a7c8ae613678fbf Mon Sep 17 00:00:00 2001
From: 雾聪 <wucong.lyb@alibaba-inc.com>
Date: 星期三, 27 三月 2024 14:11:27 +0800
Subject: [PATCH] update warmup to 10
---
runtime/onnxruntime/src/paraformer-torch.cpp | 12 ++++++++++--
1 files changed, 10 insertions(+), 2 deletions(-)
diff --git a/runtime/onnxruntime/src/paraformer-torch.cpp b/runtime/onnxruntime/src/paraformer-torch.cpp
index 934a0ea..06c88f6 100644
--- a/runtime/onnxruntime/src/paraformer-torch.cpp
+++ b/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: " << am_model << e.what();
+ exit(-1);
+ }
}
void ParaformerTorch::InitLm(const std::string &lm_file,
@@ -359,6 +366,7 @@
}
std::vector<std::vector<float>> ParaformerTorch::CompileHotwordEmbedding(std::string &hotwords) {
+ // TODO
std::vector<std::vector<float>> result(1, std::vector<float>(10, 0.0f));
return result;
}
--
Gitblit v1.9.1