Reduce model initialization time for offline speech recognition (#213)

2023-07-14 18:07:27 +08:00
parent 0abd7ce881
commit f3206c49dc
10 changed files with 109 additions and 36 deletions
--- a/sherpa-onnx/python/sherpa_onnx/offline_recognizer.py
+++ b/sherpa-onnx/python/sherpa_onnx/offline_recognizer.py
@@ -86,6 +86,7 @@ class OfflineRecognizer(object):
            num_threads=num_threads,
            debug=debug,
            provider=provider,
+            model_type="transducer",
        )

        feat_config = OfflineFeatureExtractorConfig(
@@ -149,6 +150,7 @@ class OfflineRecognizer(object):
            num_threads=num_threads,
            debug=debug,
            provider=provider,
+            model_type="paraformer",
        )

        feat_config = OfflineFeatureExtractorConfig(
@@ -211,6 +213,7 @@ class OfflineRecognizer(object):
            num_threads=num_threads,
            debug=debug,
            provider=provider,
+            model_type="nemo_ctc",
        )

        feat_config = OfflineFeatureExtractorConfig(