Add tail_paddings to Whisper C API. (#886)
This commit is contained in:
@@ -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)]
|
||||
|
||||
@@ -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))
|
||||
|
||||
|
||||
@@ -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;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user