Closed mrshu closed 3 years ago
Please use onnxruntime.transformers.convert_to_onnx
Thanks @wangyems! Does that mean the onnxruntime_tools
package is now deprecated?
Thanks @wangyems! Does that mean the
onnxruntime_tools
package is now deprecated?
Yes, if you take a look at readme: https://github.com/microsoft/onnxruntime/tree/master/onnxruntime/python/tools/transformers#gpt-2-model-conversion
Describe the bug
I believe there are two problems here:
1.6.0
version ofonnxruntime-tools
does not work withtransformers
of version>4.2
transformers
in version4.2.2
, the conversion is unsuccessful -- the command bellow finishes with non-zero exit code whilst not creating a.onnx
model:Urgency I am not sure how to assess this, but I also don't see how to currently make use of GPT2 with
onnxruntime-tools
System information
To Reproduce
All the steps to reproduce, along with all the related experiments, can be found in the following Google Colab notebook:
https://colab.research.google.com/drive/1HSYS32OtkRVJa3fShRVp_q0mxyGgpDqY
Expected behavior I would expect an
.onnx
model to be created.