EngineX-Iluvatar/enginex-mr_series-sherpa-onnx
Archived
16
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

Add Go API for Dolphin CTC models (#2090)

Browse Source
This commit is contained in:
Fangjun Kuang
2025-04-03 00:02:09 +08:00
committed by GitHub
parent 2dc0f91904
commit ba7d8b63f0
8 changed files with 36 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
scripts/go/_internal/non-streaming-decode-files/run-dolphin-ctc-base.sh Symbolic link
View File

@@ -0,0 +1 @@
../../../../go-api-examples/non-streaming-decode-files/run-dolphin-ctc-base.sh

7
scripts/go/sherpa_onnx.go
View File

@@ -377,6 +377,10 @@ type OfflineNemoEncDecCtcModelConfig struct {
Model string // Path to the model, e.g., model.onnx or model.int8.onnx
}
type OfflineDolphinModelConfig struct {
Model string // Path to the model, e.g., model.onnx or model.int8.onnx
}
type OfflineWhisperModelConfig struct {
Encoder string
Decoder string
@@ -422,6 +426,7 @@ type OfflineModelConfig struct {
SenseVoice OfflineSenseVoiceModelConfig
Moonshine OfflineMoonshineModelConfig
FireRedAsr OfflineFireRedAsrModelConfig
Dolphin OfflineDolphinModelConfig
Tokens string // Path to tokens.txt
// Number of threads to use for neural network computation
@@ -512,6 +517,8 @@ func newCOfflineRecognizerConfig(config *OfflineRecognizerConfig) *C.struct_Sher
c.model_config.fire_red_asr.encoder = C.CString(config.ModelConfig.FireRedAsr.Encoder)
c.model_config.fire_red_asr.decoder = C.CString(config.ModelConfig.FireRedAsr.Decoder)
c.model_config.dolphin.model = C.CString(config.ModelConfig.Dolphin.Model)
c.model_config.tokens = C.CString(config.ModelConfig.Tokens)
c.model_config.num_threads = C.int(config.ModelConfig.NumThreads)