Expose 'language' 'task' 'tailPaddings' in OfflineWhisperModelConfig (#643)

Co-authored-by: Gary <gary.laurenceau@gmail.com>
This commit is contained in:
GaryLaurenceauAva
2024-03-08 12:52:30 +01:00
committed by GitHub
parent 4b708e055c
commit ac43c2d7b6
2 changed files with 19 additions and 0 deletions

View File

@@ -75,6 +75,9 @@ data class OfflineParaformerModelConfig(
data class OfflineWhisperModelConfig(
var encoder: String = "",
var decoder: String = "",
var language: String = "en", // Used with multilingual model
var task: String = "transcribe", // transcribe or translate
var tailPaddings: Int = 1000, // Padding added at the end of the samples
)
data class OfflineModelConfig(