Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Export NeMo FastConformer Hybrid Transducer-CTC Large Streaming to ONNX. #843

Merged
merged 1 commit into from
May 8, 2024

Conversation

csukuangfj
Copy link
Collaborator

Fixes #790

CC @tempops @sangeet2020

Also CC @titu1994

Exported models can be found at
https://github.com/k2-fsa/sherpa-onnx/releases/tag/asr-models

Screenshot 2024-05-08 at 12 31 36

Will add

  • C++ runtime and other API bindings for it
  • Android demo
  • transducer support

@csukuangfj csukuangfj merged commit a9f936e into k2-fsa:master May 8, 2024
@csukuangfj csukuangfj deleted the nemo-fast-conformer-export branch May 8, 2024 04:33
@sangeet2020
Copy link
Contributor

Hi @csukuangfj ,

Thanks for the PR.
If I understand correctly, this PR is only meant for exporting the EncDecHybridRNNTCTCBPEModel model types such as one found in nvidia/stt_en_fastconformer_hybrid_large_streaming_multi.

The decoding implementation needs to be done separately? or can I still use the existing implementation https://k2-fsa.github.io/sherpa/onnx/pretrained_models/offline-ctc/nemo/index.html for these models?

@csukuangfj
Copy link
Collaborator Author

If I understand correctly, this PR is only meant for exporting the EncDecHybridRNNTCTCBPEModel model types such as one found in nvidia/stt_en_fastconformer_hybrid_large_streaming_multi.

Yes, you are right.


The decoding implementation needs to be done separately?

No, we can reuse the existing CTC decoding code.

However, we need to change the C++ code to support the neural network part.

By the way, the C++ decoding part does not depend on the specific neural network model; therefore, it can be reused for any model trained by the CTC loss.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Help for exporting FastConformer NeMo model to onnx for use in sherpa-onnx for streaming inference
2 participants