This commit is contained in:
Fangjun Kuang
2024-06-04 17:05:49 +08:00
committed by GitHub
parent c6f22f25fa
commit f8dbc10146
9 changed files with 123 additions and 10 deletions

View File

@@ -88,7 +88,9 @@ func sherpaOnnxOnlineModelConfig(
numThreads: Int = 1,
provider: String = "cpu",
debug: Int = 0,
modelType: String = ""
modelType: String = "",
modelingUnit: String = "cjkchar",
bpeVocab: String = ""
) -> SherpaOnnxOnlineModelConfig {
return SherpaOnnxOnlineModelConfig(
transducer: transducer,
@@ -98,7 +100,9 @@ func sherpaOnnxOnlineModelConfig(
num_threads: Int32(numThreads),
provider: toCPointer(provider),
debug: Int32(debug),
model_type: toCPointer(modelType)
model_type: toCPointer(modelType),
modeling_unit: toCPointer(modelingUnit),
bpeVocab: toCPointer(bpeVocab)
)
}
@@ -354,7 +358,9 @@ func sherpaOnnxOfflineModelConfig(
numThreads: Int = 1,
provider: String = "cpu",
debug: Int = 0,
modelType: String = ""
modelType: String = "",
modelingUnit: String = "cjkchar",
bpeVocab: String = ""
) -> SherpaOnnxOfflineModelConfig {
return SherpaOnnxOfflineModelConfig(
transducer: transducer,
@@ -366,7 +372,9 @@ func sherpaOnnxOfflineModelConfig(
num_threads: Int32(numThreads),
debug: Int32(debug),
provider: toCPointer(provider),
model_type: toCPointer(modelType)
model_type: toCPointer(modelType),
modeling_unit: toCPointer(modelingUnit),
bpeVocab: toCPointer(bpeVocab)
)
}