游雁
2023-04-23 6d8362f80beb3ef13af5312fc2c74a9d6ecf308a
funasr/runtime/python/onnxruntime/funasr_onnx/paraformer_bin.py
@@ -19,6 +19,11 @@
class Paraformer():
    """
    Author: Speech Lab of DAMO Academy, Alibaba Group
    Paraformer: Fast and Accurate Parallel Transformer for Non-autoregressive End-to-End Speech Recognition
    https://arxiv.org/abs/2206.08317
    """
    def __init__(self, model_dir: Union[str, Path] = None,
                 batch_size: int = 1,
                 device_id: Union[str, int] = "-1",