From 200bba87f0060cfbc92c6ff45026313c188b6452 Mon Sep 17 00:00:00 2001 From: 雾聪 <wucong.lyb@alibaba-inc.com> Date: 星期三, 17 四月 2024 19:14:20 +0800 Subject: [PATCH] sort frame_queue --- runtime/onnxruntime/include/com-define.h | 8 ++++++++ 1 files changed, 8 insertions(+), 0 deletions(-) diff --git a/runtime/onnxruntime/include/com-define.h b/runtime/onnxruntime/include/com-define.h index 9cb1f2c..9480b0b 100644 --- a/runtime/onnxruntime/include/com-define.h +++ b/runtime/onnxruntime/include/com-define.h @@ -49,6 +49,14 @@ // hotword embedding compile model #define MODEL_EB_NAME "model_eb.onnx" #define QUANT_MODEL_NAME "model_quant.onnx" +// 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 BLADEDISC "bladedisc" + #define VAD_CMVN_NAME "vad.mvn" #define VAD_CONFIG_NAME "vad.yaml" #define AM_CMVN_NAME "am.mvn" -- Gitblit v1.9.1