MahmoudAshraf97 / ctc-forced-aligner

Text to speech alignment using CTC forced alignment
150 stars 31 forks source link

MahmoudAshraf/mms-300m-1130-forced-aligner Support ONNX Model #33

Closed jc955 closed 2 weeks ago

jc955 commented 2 weeks ago

hi 👋 I want to run the ONNX model version of MahmoudAshraf/mms-300m-1130-forced-aligner on onnxruntime-node. Is there a way to perform model conversion?

MahmoudAshraf97 commented 2 weeks ago

You can check HF optimum library, it provides tools to export HF models to onnx You'll then replace the following lines with the new onnx inference: https://github.com/MahmoudAshraf97/ctc-forced-aligner/blob/c7cc7ce609e5f8f1f553fbd1e53124447ffe46d8/ctc_forced_aligner/alignment_utils.py#L134-L138