search

xiaomingnio / kantts

TTS appalication based on modelscope KAN-TTS
43 stars 6 forks source link

sambert导出onnx #3

Closed Leslie-Tsui closed 10 months ago

Leslie-Tsui commented 10 months ago

作者您好，请问有sambert导出onnx的脚步吗？

xiaomingnio commented 10 months ago

sambert 无法导出onnx，目前只能用tensorrt api重写sambert部分代码进行加速。

Leslie-Tsui commented 10 months ago

sambert 无法导出onnx，目前只能用tensorrt api重写sambert部分代码进行加速。

谢谢，但我对这方面的工作不是很熟悉，请问能提供tensorrt重写的代码吗？

xiaomingnio commented 10 months ago

后面有时间会更新完整代码

xiaomingnio commented 10 months ago

https://github.com/xiaomingnio/kantts/blob/main/tensorrt_onnx/build_from_network.py