chatterbox icon indicating copy to clipboard operation
chatterbox copied to clipboard

Onnx model

Open thewh1teagle opened this issue 6 months ago • 9 comments

Hi, thank you for amazing work! I would like to use the model with onnxruntime, so I won't need to install heavy dependencies. How can I use it with that?

thewh1teagle avatar May 29 '25 09:05 thewh1teagle

That would be very useful. Especially for running it on different machines. Onnx also has webgpu integration recently, which is pretty cool for a cross-platform experience. :-)

I've been looking into the codebase but its obvious quite complex. core TTS likely involves the T3 and S3Gen (sep models). For an ONNX conversion, which would likely require model splitting, I'd be grateful for any guidance from the maintainers, particularly on strategies for handling the dynamic control flow of the AlignmentStreamAnalyzer, the iterative ODE solving loop within S3Gen's ConditionalCFM, and any custom operations like SineGen or the Snake activations. Any pointers on how these could be made ONNX-compatible would be a fantastic

Thanks for this great work!

altunenes avatar May 29 '25 14:05 altunenes

Very interested in this too! 🤗

Edit: @VladOS95-cyber and I got it working!

  • English-only: https://huggingface.co/onnx-community/chatterbox-ONNX

  • Multilingual: https://huggingface.co/onnx-community/chatterbox-multilingual-ONNX

Enjoy!

xenova avatar May 30 '25 23:05 xenova

+1

babybirdprd avatar Jun 05 '25 03:06 babybirdprd

I privately attempted an ONNX migration. I was able to get each of the models ported, apart from the VO Encoder.

nickheyer avatar Jun 05 '25 18:06 nickheyer

+1

JacobRyanHanson avatar Jul 09 '25 11:07 JacobRyanHanson

I privately attempted an ONNX migration. I was able to get each of the models ported, apart from the VO Encoder.

@nickheyer hey buddy i am eagerly looking for onnx option for chatterbox. it would be really great if you could share your findings. thanks in advance

riteshraav avatar Jul 14 '25 13:07 riteshraav

+1

saikumar0605 avatar Jul 18 '25 15:07 saikumar0605

+1

SkutteOleg avatar Sep 06 '25 06:09 SkutteOleg

For anyone still needing the ONNX model: https://huggingface.co/onnx-community/chatterbox-multilingual-ONNX

Non-multilingual Model: https://huggingface.co/onnx-community/chatterbox-ONNX

They seem to work.

mimouza23 avatar Oct 09 '25 18:10 mimouza23