How to use export-onnx.py to change the pytorch_model.bin to onnx?

Beginners
1

image
image630×453 6.82 KB

image
image381×241 7.88 KB

I want to change the fine-tuned whisper model to onnx model, but i don’t know how to finish it, the export-onnx.py have the main() function , it only support the type of the fix form which it set in the files. I want to change my own model to onnx, so i need some help.

1 Like
2

Whisper seems to be a little special, and it seems to require a few steps. I think Optimum’s ONNX Exporter is newer, but I don’t know if it supports customized Whisper.

WhisperのモデルをONNXにする - TadaoYamaokaの開発日記 # in Japanese language

1 Like