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> CompileHotwordEmbedding(std::string &hotwords){return std::vector>();}; 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;};