diff --git a/swift-api-examples/SherpaOnnx.swift b/swift-api-examples/SherpaOnnx.swift index 9bb3794b..7310d94f 100644 --- a/swift-api-examples/SherpaOnnx.swift +++ b/swift-api-examples/SherpaOnnx.swift @@ -35,6 +35,7 @@ func sherpaOnnxOnlineTransducerModelConfig( joiner: String, tokens: String, numThreads: Int = 2, + provider: String = "cpu", debug: Int = 0 ) -> SherpaOnnxOnlineTransducerModelConfig { return SherpaOnnxOnlineTransducerModelConfig( @@ -43,6 +44,7 @@ func sherpaOnnxOnlineTransducerModelConfig( joiner: toCPointer(joiner), tokens: toCPointer(tokens), num_threads: Int32(numThreads), + provider: toCPointer(provider), debug: Int32(debug) ) }