Add tail_paddings to Whisper C API. (#886)

This commit is contained in:
Fangjun Kuang
2024-05-17 09:20:07 +08:00
committed by GitHub
parent 65635b09d8
commit 8af2af8466
13 changed files with 29 additions and 9 deletions

View File

@@ -336,10 +336,11 @@ type OfflineNemoEncDecCtcModelConfig struct {
}
type OfflineWhisperModelConfig struct {
Encoder string
Decoder string
Language string
Task string
Encoder string
Decoder string
Language string
Task string
TailPaddings int
}
type OfflineTdnnModelConfig struct {
@@ -441,6 +442,8 @@ func NewOfflineRecognizer(config *OfflineRecognizerConfig) *OfflineRecognizer {
c.model_config.whisper.task = C.CString(config.ModelConfig.Whisper.Task)
defer C.free(unsafe.Pointer(c.model_config.whisper.task))
c.model_config.whisper.tail_paddings = C.int(config.ModelConfig.Whisper.TailPaddings)
c.model_config.tdnn.model = C.CString(config.ModelConfig.Tdnn.Model)
defer C.free(unsafe.Pointer(c.model_config.tdnn.model))