From b2d549b9541bd50dc744a152ddaca090bb254a9f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=9B=BE=E8=81=AA?= <wucong.lyb@alibaba-inc.com>
Date: Fri, 29 Mar 2024 17:06:35 +0800
Subject: [PATCH] add SetBatchSize for model

---
 runtime/onnxruntime/include/model.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/runtime/onnxruntime/include/model.h b/runtime/onnxruntime/include/model.h
index 8b84abb17..b8eb49137 100644
--- a/runtime/onnxruntime/include/model.h
+++ b/runtime/onnxruntime/include/model.h
@@ -30,7 +30,7 @@ class Model {
     virtual std::vector<std::vector<float>> CompileHotwordEmbedding(std::string &hotwords){return std::vector<std::vector<float>>();};
     virtual std::string GetLang(){return "";};
     virtual int GetAsrSampleRate() = 0;
-    virtual void SetBatchSize() {};
+    virtual void SetBatchSize(int batch_size) {};
     virtual int GetBatchSize() {return 0;};
     virtual Vocab* GetVocab() {return nullptr;};
     virtual Vocab* GetLmVocab() {return nullptr;};