modelscope/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR.git synced 2026-10-12 02:19:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

add SetBatchSize for model

Browse Source
This commit is contained in:
雾聪
2024-03-29 17:06:35 +08:00
parent 3ac539a58c
commit b2d549b954
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
runtime/onnxruntime/include/model.h
+1 -1
View File
@@ -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;};