@@ -29,10 +29,15 @@ data class OnlineZipformer2CtcModelConfig(
29
29
var model : String = " " ,
30
30
)
31
31
32
+ data class OnlineNeMoCtcModelConfig (
33
+ var model : String = " " ,
34
+ )
35
+
32
36
data class OnlineModelConfig (
33
37
var transducer : OnlineTransducerModelConfig = OnlineTransducerModelConfig (),
34
38
var paraformer : OnlineParaformerModelConfig = OnlineParaformerModelConfig (),
35
39
var zipformer2Ctc : OnlineZipformer2CtcModelConfig = OnlineZipformer2CtcModelConfig (),
40
+ var neMoCtc : OnlineNeMoCtcModelConfig = OnlineNeMoCtcModelConfig (),
36
41
var tokens : String ,
37
42
var numThreads : Int = 1 ,
38
43
var debug : Boolean = false ,
@@ -318,6 +323,36 @@ fun getModelConfig(type: Int): OnlineModelConfig? {
318
323
modelType = " zipformer" ,
319
324
)
320
325
}
326
+
327
+ 11 -> {
328
+ val modelDir = " sherpa-onnx-nemo-streaming-fast-conformer-ctc-en-80ms"
329
+ return OnlineModelConfig (
330
+ neMoCtc = OnlineNeMoCtcModelConfig (
331
+ model = " $modelDir /model.onnx" ,
332
+ ),
333
+ tokens = " $modelDir /tokens.txt" ,
334
+ )
335
+ }
336
+
337
+ 12 -> {
338
+ val modelDir = " sherpa-onnx-nemo-streaming-fast-conformer-ctc-en-480ms"
339
+ return OnlineModelConfig (
340
+ neMoCtc = OnlineNeMoCtcModelConfig (
341
+ model = " $modelDir /model.onnx" ,
342
+ ),
343
+ tokens = " $modelDir /tokens.txt" ,
344
+ )
345
+ }
346
+
347
+ 13 -> {
348
+ val modelDir = " sherpa-onnx-nemo-streaming-fast-conformer-ctc-en-1040ms"
349
+ return OnlineModelConfig (
350
+ neMoCtc = OnlineNeMoCtcModelConfig (
351
+ model = " $modelDir /model.onnx" ,
352
+ ),
353
+ tokens = " $modelDir /tokens.txt" ,
354
+ )
355
+ }
321
356
}
322
357
return null
323
358
}
0 commit comments