From 7f51f6108eb0c92fa01d0bfc3592fbb8fc8ee838 Mon Sep 17 00:00:00 2001
From: 雾聪 <wucong.lyb@alibaba-inc.com>
Date: 星期二, 26 三月 2024 16:02:36 +0800
Subject: [PATCH] add bladedisc for funasr-onnx-offline
---
runtime/onnxruntime/src/paraformer-torch.cpp | 3 ++-
1 files changed, 2 insertions(+), 1 deletions(-)
diff --git a/runtime/onnxruntime/src/paraformer-torch.cpp b/runtime/onnxruntime/src/paraformer-torch.cpp
index a08f1bd..bdfd0ee 100644
--- a/runtime/onnxruntime/src/paraformer-torch.cpp
+++ b/runtime/onnxruntime/src/paraformer-torch.cpp
@@ -38,7 +38,7 @@
LOG(ERROR) << "CUDA is not available! Please check your GPU settings";
exit(-1);
} else {
- LOG(INFO) << "CUDA available! Running on GPU";
+ LOG(INFO) << "CUDA is available, running on GPU";
device = at::kCUDA;
}
#endif
@@ -359,6 +359,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