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

Add C++ runtime for non-streaming faster conformer transducer from NeMo. (#854)

Browse Source
This commit is contained in:
Fangjun Kuang
2024-05-10 12:15:39 +08:00
committed by GitHub
parent 5d8c35e44e
commit 17cd3a5f01
31 changed files with 1093 additions and 153 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
sherpa-onnx/python/sherpa_onnx/offline_recognizer.py
View File

@@ -4,8 +4,8 @@ from pathlib import Path
from typing import List, Optional
from _sherpa_onnx import (
FeatureExtractorConfig,
OfflineCtcFstDecoderConfig,
OfflineFeatureExtractorConfig,
OfflineModelConfig,
OfflineNemoEncDecCtcModelConfig,
OfflineParaformerModelConfig,
@@ -51,6 +51,7 @@ class OfflineRecognizer(object):
blank_penalty: float = 0.0,
debug: bool = False,
provider: str = "cpu",
model_type: str = "transducer",
):
"""
Please refer to
@@ -106,10 +107,10 @@ class OfflineRecognizer(object):
num_threads=num_threads,
debug=debug,
provider=provider,
model_type="transducer",
model_type=model_type,
)
feat_config = OfflineFeatureExtractorConfig(
feat_config = FeatureExtractorConfig(
sampling_rate=sample_rate,
feature_dim=feature_dim,
)
@@ -182,7 +183,7 @@ class OfflineRecognizer(object):
model_type="paraformer",
)
feat_config = OfflineFeatureExtractorConfig(
feat_config = FeatureExtractorConfig(
sampling_rate=sample_rate,
feature_dim=feature_dim,
)
@@ -246,7 +247,7 @@ class OfflineRecognizer(object):
model_type="nemo_ctc",
)
feat_config = OfflineFeatureExtractorConfig(
feat_config = FeatureExtractorConfig(
sampling_rate=sample_rate,
feature_dim=feature_dim,
)
@@ -326,7 +327,7 @@ class OfflineRecognizer(object):
model_type="whisper",
)
feat_config = OfflineFeatureExtractorConfig(
feat_config = FeatureExtractorConfig(
sampling_rate=16000,
feature_dim=80,
)
@@ -389,7 +390,7 @@ class OfflineRecognizer(object):
model_type="tdnn",
)
feat_config = OfflineFeatureExtractorConfig(
feat_config = FeatureExtractorConfig(
sampling_rate=sample_rate,
feature_dim=feature_dim,
)
@@ -453,7 +454,7 @@ class OfflineRecognizer(object):
model_type="wenet_ctc",
)
feat_config = OfflineFeatureExtractorConfig(
feat_config = FeatureExtractorConfig(
sampling_rate=sample_rate,
feature_dim=feature_dim,
)