From 28ccfbfc51068a663a80764e14074df5edf2b5ba Mon Sep 17 00:00:00 2001 From: kongdeqiang <kongdeqiang960204@163.com> Date: 星期五, 13 三月 2026 17:41:41 +0800 Subject: [PATCH] 提交 --- runtime/onnxruntime/include/com-define.h | 19 ++++++++++++++----- 1 files changed, 14 insertions(+), 5 deletions(-) diff --git a/runtime/onnxruntime/include/com-define.h b/runtime/onnxruntime/include/com-define.h index 70e1495..6b4c08f 100644 --- a/runtime/onnxruntime/include/com-define.h +++ b/runtime/onnxruntime/include/com-define.h @@ -36,6 +36,9 @@ #define HOTWORD_SEP " " #define AUDIO_FS "audio-fs" +#define MODEL_PARA "Paraformer" +#define MODEL_SVS "SenseVoiceSmall" + // #define VAD_MODEL_PATH "vad-model" // #define VAD_CMVN_PATH "vad-cmvn" // #define VAD_CONFIG_PATH "vad-config" @@ -48,19 +51,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