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

@@ -301,6 +301,7 @@ namespace SherpaOnnx
Decoder = "";
Language = "";
Task = "transcribe";
TailPaddings = -1;
}
[MarshalAs(UnmanagedType.LPStr)]
public string Encoder;
@@ -313,6 +314,8 @@ namespace SherpaOnnx
[MarshalAs(UnmanagedType.LPStr)]
public string Task;
public int TailPaddings;
}
[StructLayout(LayoutKind.Sequential)]

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))

View File

@@ -74,7 +74,8 @@ static SherpaOnnxOfflineWhisperModelConfig GetOfflineWhisperModelConfig(
SHERPA_ONNX_ASSIGN_ATTR_STR(encoder, encoder);
SHERPA_ONNX_ASSIGN_ATTR_STR(decoder, decoder);
SHERPA_ONNX_ASSIGN_ATTR_STR(language, language);
SHERPA_ONNX_ASSIGN_ATTR_STR(task, languagek);
SHERPA_ONNX_ASSIGN_ATTR_STR(task, task);
SHERPA_ONNX_ASSIGN_ATTR_INT32(tail_paddings, tailPaddings);
return c;
}