update comdefine

雾聪

2024-03-26 423c29d7f9a6bf8711dc50231d2bfc28f7cc0099

update comdefine

 runtime/onnxruntime/include/com-define.h

@@ -51,8 +51,10 @@
#define QUANT_MODEL_NAME "model_quant.onnx"
// gpu models
#define INFER_GPU "gpu"
#define TORCH_MODEL_NAME "model.gpu.torchscripts"
#define TORCH_QUANT_MODEL_NAME "model.blade.fp16.pt"
#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"

 runtime/onnxruntime/src/offline-stream.cpp

@@ -70,6 +70,9 @@
            if(model_path.find(QUANTIZE) != model_path.end() && model_path.at(QUANTIZE) == "true"){
                am_model_path = PathAppend(model_path.at(MODEL_DIR), TORCH_QUANT_MODEL_NAME);
            }
            if(model_path.find(BLADEDISC) != model_path.end() && model_path.at(BLADEDISC) == "true"){
                am_model_path = PathAppend(model_path.at(MODEL_DIR), BLADE_MODEL_NAME);
            }
          }
        }
        am_cmvn_path = PathAppend(model_path.at(MODEL_DIR), AM_CMVN_NAME);