optimum
874d2b25 - Add argument in the CLI to specify device to do the ONNX export on (#634)

Commit
2 years ago
Add argument in the CLI to specify device to do the ONNX export on (#634) * add args device * do validation on device as well * handle device for list and tuple * fix test * Update optimum/exporters/onnx/convert.py Co-authored-by: Michael Benayoun <mickbenayoun@gmail.com> * Update optimum/exporters/onnx/convert.py Co-authored-by: Michael Benayoun <mickbenayoun@gmail.com> * fix test * fix torch import * fix import2 * debug * run all Co-authored-by: Michael Benayoun <mickbenayoun@gmail.com>
Author
Parents
Loading