From 172a3152b42af36443ec6a0a39969471c35b893d Mon Sep 17 00:00:00 2001
From: maliubiao <maliubiao@gmail.com>
Date: 星期日, 29 十二月 2024 22:33:22 +0800
Subject: [PATCH] 允许 model.generate 使用bytes io, 以便不写入文件，节省io时间 (#2343)
---
runtime/onnxruntime/include/vad-model.h | 1 +
1 files changed, 1 insertions(+), 0 deletions(-)
diff --git a/runtime/onnxruntime/include/vad-model.h b/runtime/onnxruntime/include/vad-model.h
index 07f1833..adb1e20 100644
--- a/runtime/onnxruntime/include/vad-model.h
+++ b/runtime/onnxruntime/include/vad-model.h
@@ -12,6 +12,7 @@
virtual ~VadModel(){};
virtual void InitVad(const std::string &vad_model, const std::string &vad_cmvn, const std::string &vad_config, int thread_num)=0;
virtual std::vector<std::vector<int>> Infer(std::vector<float> &waves, bool input_finished=true)=0;
+ virtual int GetVadSampleRate() = 0;
};
VadModel *CreateVadModel(std::map<std::string, std::string>& model_path, int thread_num);
--
Gitblit v1.9.1