From d80ac2fd2df4e7fb8a28acfa512bb11472b5cc99 Mon Sep 17 00:00:00 2001 From: liugz18 <57401541+liugz18@users.noreply.github.com> Date: 星期四, 18 七月 2024 21:34:55 +0800 Subject: [PATCH] Rename 'res' in line 514 to avoid with naming conflict with line 365 --- runtime/onnxruntime/include/com-define.h | 16 +++++++++++----- 1 files changed, 11 insertions(+), 5 deletions(-) diff --git a/runtime/onnxruntime/include/com-define.h b/runtime/onnxruntime/include/com-define.h index 70e1495..5f71f7b 100644 --- a/runtime/onnxruntime/include/com-define.h +++ b/runtime/onnxruntime/include/com-define.h @@ -48,19 +48,25 @@ #define MODEL_NAME "model.onnx" // hotword embedding compile model #define MODEL_EB_NAME "model_eb.onnx" +#define TORCH_MODEL_EB_NAME "model_eb.torchscript" #define QUANT_MODEL_NAME "model_quant.onnx" +#define VAD_CMVN_NAME "am.mvn" +#define VAD_CONFIG_NAME "config.yaml" + // gpu models #define INFER_GPU "gpu" -#define TORCH_MODEL_NAME "model.gpu.torchscripts" -#define TORCH_QUANT_MODEL_NAME "model.blade.fp16.pt" +#define BATCHSIZE "batch-size" +#define TORCH_MODEL_NAME "model.torchscript" +#define TORCH_QUANT_MODEL_NAME "model_quant.torchscript" +#define BLADE_MODEL_NAME "model_blade.torchscript" +#define BLADEDISC "bladedisc" -#define VAD_CMVN_NAME "vad.mvn" -#define VAD_CONFIG_NAME "vad.yaml" #define AM_CMVN_NAME "am.mvn" #define AM_CONFIG_NAME "config.yaml" #define LM_CONFIG_NAME "config.yaml" -#define PUNC_CONFIG_NAME "punc.yaml" +#define PUNC_CONFIG_NAME "config.yaml" #define MODEL_SEG_DICT "seg_dict" +#define TOKEN_PATH "tokens.json" #define HOTWORD "hotword" // #define NN_HOTWORD "nn-hotword" -- Gitblit v1.9.1