diff --git a/scripts/apk/generate-asr-apk-script.py b/scripts/apk/generate-asr-apk-script.py index 09a8915e..8684877c 100755 --- a/scripts/apk/generate-asr-apk-script.py +++ b/scripts/apk/generate-asr-apk-script.py @@ -209,6 +209,27 @@ def get_models(): ls -lh + popd + """, + ), + Model( + model_name="sherpa-onnx-streaming-zipformer-korean-2024-06-16", + idx=14, + lang="ko", + short_name="zipformer", + cmd=""" + pushd $model_name + rm -fv decoder-epoch-99-avg-1.int8.onnx + rm -fv encoder-epoch-99-avg-1.onnx + rm -fv joiner-epoch-99-avg-1.onnx + + rm -fv bpe.model + rm -fv README.md + rm -fv .gitattributes + rm -rfv test_wavs + + ls -lh + popd """, ), diff --git a/sherpa-onnx/csrc/online-recognizer-transducer-impl.h b/sherpa-onnx/csrc/online-recognizer-transducer-impl.h index 16c44b9d..a2531b10 100644 --- a/sherpa-onnx/csrc/online-recognizer-transducer-impl.h +++ b/sherpa-onnx/csrc/online-recognizer-transducer-impl.h @@ -364,7 +364,7 @@ class OnlineRecognizerTransducerImpl : public OnlineRecognizerImpl { } // reset encoder states - s->SetStates(model_->GetEncoderInitStates()); + // s->SetStates(model_->GetEncoderInitStates()); // we keep the decoder_out decoder_->UpdateDecoderOut(&s->GetResult()); diff --git a/sherpa-onnx/kotlin-api/OnlineRecognizer.kt b/sherpa-onnx/kotlin-api/OnlineRecognizer.kt index e78fb654..93a21e07 100644 --- a/sherpa-onnx/kotlin-api/OnlineRecognizer.kt +++ b/sherpa-onnx/kotlin-api/OnlineRecognizer.kt @@ -355,6 +355,19 @@ fun getModelConfig(type: Int): OnlineModelConfig? { tokens = "$modelDir/tokens.txt", ) } + + 14 -> { + val modelDir = "sherpa-onnx-streaming-zipformer-korean-2024-06-16" + return OnlineModelConfig( + transducer = OnlineTransducerModelConfig( + encoder = "$modelDir/encoder-epoch-99-avg-1.int8.onnx", + decoder = "$modelDir/decoder-epoch-99-avg-1.onnx", + joiner = "$modelDir/joiner-epoch-99-avg-1.int8.onnx", + ), + tokens = "$modelDir/tokens.txt", + modelType = "zipformer", + ) + } } return null }