From b7060884fa4b8b85f79462644a5c99062d223da0 Mon Sep 17 00:00:00 2001 From: Yabin Li <wucong.lyb@alibaba-inc.com> Date: 星期二, 25 六月 2024 17:38:04 +0800 Subject: [PATCH] Merge Dev tclas (#1847) --- runtime/onnxruntime/include/com-define.h | 7 ++++--- 1 files changed, 4 insertions(+), 3 deletions(-) diff --git a/runtime/onnxruntime/include/com-define.h b/runtime/onnxruntime/include/com-define.h index 77a7b02..5f71f7b 100644 --- a/runtime/onnxruntime/include/com-define.h +++ b/runtime/onnxruntime/include/com-define.h @@ -48,6 +48,7 @@ #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" @@ -55,9 +56,9 @@ // gpu models #define INFER_GPU "gpu" #define BATCHSIZE "batch-size" -#define TORCH_MODEL_NAME "model.torchscripts" -#define TORCH_QUANT_MODEL_NAME "model_quant.torchscripts" -#define BLADE_MODEL_NAME "model.blade.fp16.pt" +#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 AM_CMVN_NAME "am.mvn" -- Gitblit v1.9.1