正在显示
1 个修改的文件
包含
2 行增加
和
0 行删除
| @@ -35,6 +35,7 @@ func sherpaOnnxOnlineTransducerModelConfig( | @@ -35,6 +35,7 @@ func sherpaOnnxOnlineTransducerModelConfig( | ||
| 35 | joiner: String, | 35 | joiner: String, |
| 36 | tokens: String, | 36 | tokens: String, |
| 37 | numThreads: Int = 2, | 37 | numThreads: Int = 2, |
| 38 | + provider: String = "cpu", | ||
| 38 | debug: Int = 0 | 39 | debug: Int = 0 |
| 39 | ) -> SherpaOnnxOnlineTransducerModelConfig { | 40 | ) -> SherpaOnnxOnlineTransducerModelConfig { |
| 40 | return SherpaOnnxOnlineTransducerModelConfig( | 41 | return SherpaOnnxOnlineTransducerModelConfig( |
| @@ -43,6 +44,7 @@ func sherpaOnnxOnlineTransducerModelConfig( | @@ -43,6 +44,7 @@ func sherpaOnnxOnlineTransducerModelConfig( | ||
| 43 | joiner: toCPointer(joiner), | 44 | joiner: toCPointer(joiner), |
| 44 | tokens: toCPointer(tokens), | 45 | tokens: toCPointer(tokens), |
| 45 | num_threads: Int32(numThreads), | 46 | num_threads: Int32(numThreads), |
| 47 | + provider: toCPointer(provider), | ||
| 46 | debug: Int32(debug) | 48 | debug: Int32(debug) |
| 47 | ) | 49 | ) |
| 48 | } | 50 | } |
-
请 注册 或 登录 后发表评论