From 3ac539a58ca56a04123771ef145430cbcbdc294e 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:04:39 +0800
Subject: [PATCH] add SetBatchSize for model

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

diff --git a/runtime/onnxruntime/include/model.h b/runtime/onnxruntime/include/model.h
index 9fec8fa7d..8b84abb17 100644
--- a/runtime/onnxruntime/include/model.h
+++ b/runtime/onnxruntime/include/model.h
@@ -30,6 +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 int GetBatchSize() {return 0;};
     virtual Vocab* GetVocab() {return nullptr;};
     virtual Vocab* GetLmVocab() {return nullptr;};