From 00cfc36b9a1ad4d114434eb7770c1e67940d4862 Mon Sep 17 00:00:00 2001 From: Yabin Li <wucong.lyb@alibaba-inc.com> Date: 星期一, 13 五月 2024 16:52:41 +0800 Subject: [PATCH] c++ runtime adapt to 1.0 (#1724) --- runtime/onnxruntime/src/ct-transformer.h | 2 +- 1 files changed, 1 insertions(+), 1 deletions(-) diff --git a/runtime/onnxruntime/src/ct-transformer.h b/runtime/onnxruntime/src/ct-transformer.h index b33dcf5..f38fe12 100644 --- a/runtime/onnxruntime/src/ct-transformer.h +++ b/runtime/onnxruntime/src/ct-transformer.h @@ -26,7 +26,7 @@ public: CTTransformer(); - void InitPunc(const std::string &punc_model, const std::string &punc_config, int thread_num); + void InitPunc(const std::string &punc_model, const std::string &punc_config, const std::string &token_file, int thread_num); ~CTTransformer(); vector<int> Infer(vector<int32_t> input_data); string AddPunc(const char* sz_input, std::string language="zh-cn"); -- Gitblit v1.9.1