add SetBatchSize for model

This commit is contained in:
雾聪 2024-03-29 17:04:39 +08:00
parent cc10d01fd7
commit 3ac539a58c

View File

@ -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;};