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>