extern/whisper.cpp
1
0
Fork 1
mirror of https://github.com/ggerganov/whisper.cpp.git synced 2024-12-28 09:39:07 +01:00
Code Issues Packages Projects Releases Wiki Activity

go : implement SetSplitOnWord (#1114)

Browse Source 
* Go binding: Implement SetSplitOnWord

* Add comment for consistency
This commit is contained in:
xdrudis 2023-07-25 09:10:12 -07:00 committed by GitHub
parent 1450346214
commit a2684cd93a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
bindings/go/params.go
View File

@ -19,6 +19,10 @@ func (p *Params) SetTranslate(v bool) {
p.translate = toBool(v) p.translate = toBool(v)
} }
func (p *Params) SetSplitOnWord(v bool) {
p.split_on_word = toBool(v)
}
func (p *Params) SetNoContext(v bool) { func (p *Params) SetNoContext(v bool) {
p.no_context = toBool(v) p.no_context = toBool(v)
} }

4
bindings/go/pkg/whisper/context.go
View File

@ -81,6 +81,10 @@ func (context *context) SetSpeedup(v bool) {
context.params.SetSpeedup(v) context.params.SetSpeedup(v)
} }
func (context *context) SetSplitOnWord(v bool) {
context.params.SetSplitOnWord(v)
}
// Set number of threads to use // Set number of threads to use
func (context *context) SetThreads(v uint) { func (context *context) SetThreads(v uint) {
context.params.SetThreads(int(v)) context.params.SetThreads(int(v))

1
bindings/go/pkg/whisper/interface.go
View File

@ -42,6 +42,7 @@ type Context interface {
SetDuration(time.Duration) // Set duration SetDuration(time.Duration) // Set duration
SetThreads(uint) // Set number of threads to use SetThreads(uint) // Set number of threads to use
SetSpeedup(bool) // Set speedup flag SetSpeedup(bool) // Set speedup flag
SetSplitOnWord(bool) // Set split on word flag
SetTokenThreshold(float32) // Set timestamp token probability threshold SetTokenThreshold(float32) // Set timestamp token probability threshold
SetTokenSumThreshold(float32) // Set timestamp token sum probability threshold SetTokenSumThreshold(float32) // Set timestamp token sum probability threshold
SetMaxSegmentLength(uint) // Set max segment length in characters SetMaxSegmentLength(uint) // Set max segment length in characters