EngineX-Iluvatar/enginex_bi_series-sherpa-onnx
Archived
16
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

Fix CI errors introduced by supporting loading keywords from buffers (#1366)

Browse Source
This commit is contained in:
Fangjun Kuang
2024-09-20 19:04:21 +08:00
committed by GitHub
parent 06b61ccad8
commit d8809b520e
8 changed files with 55 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
flutter/sherpa_onnx/lib/src/keyword_spotter.dart
View File

@@ -19,11 +19,13 @@ class KeywordSpotterConfig {
this.keywordsScore = 1.0,
this.keywordsThreshold = 0.25,
this.keywordsFile = '',
this.keywordsBuf = '',
this.keywordsBufSize = 0,
});
@override
String toString() {
return 'KeywordSpotterConfig(feat: $feat, model: $model, maxActivePaths: $maxActivePaths, numTrailingBlanks: $numTrailingBlanks, keywordsScore: $keywordsScore, keywordsThreshold: $keywordsThreshold, keywordsFile: $keywordsFile)';
return 'KeywordSpotterConfig(feat: $feat, model: $model, maxActivePaths: $maxActivePaths, numTrailingBlanks: $numTrailingBlanks, keywordsScore: $keywordsScore, keywordsThreshold: $keywordsThreshold, keywordsFile: $keywordsFile, keywordsBuf: $keywordsBuf, keywordsBufSize: $keywordsBufSize)';
}
final FeatureConfig feat;
@@ -35,6 +37,8 @@ class KeywordSpotterConfig {
final double keywordsScore;
final double keywordsThreshold;
final String keywordsFile;
final String keywordsBuf;
final int keywordsBufSize;
}
class KeywordResult {
@@ -89,9 +93,12 @@ class KeywordSpotter {
c.ref.keywordsScore = config.keywordsScore;
c.ref.keywordsThreshold = config.keywordsThreshold;
c.ref.keywordsFile = config.keywordsFile.toNativeUtf8();
c.ref.keywordsBuf = config.keywordsBuf.toNativeUtf8();
c.ref.keywordsBufSize = config.keywordsBufSize;
final ptr = SherpaOnnxBindings.createKeywordSpotter?.call(c) ?? nullptr;
calloc.free(c.ref.keywordsBuf);
calloc.free(c.ref.keywordsFile);
calloc.free(c.ref.model.bpeVocab);
calloc.free(c.ref.model.modelingUnit);

5
flutter/sherpa_onnx/lib/src/sherpa_onnx_bindings.dart
View File

@@ -371,6 +371,11 @@ final class SherpaOnnxKeywordSpotterConfig extends Struct {
external double keywordsThreshold;
external Pointer<Utf8> keywordsFile;
external Pointer<Utf8> keywordsBuf;
@Int32()
external int keywordsBufSize;
}
final class SherpaOnnxOfflinePunctuation extends Opaque {}