optimum
Add argument in the CLI to specify device to do the ONNX export on
#634
Merged
Overview
Commits
11
Changes
View On GitHub

Add argument in the CLI to specify device to do the ONNX export on #634

fxmarty merged 11 commits into huggingface:main from fxmarty:authorize-custom-device-exporters-onnx
fxmarty
fxmarty add args device
9e0d4da0
fxmarty do validation on device as well
a79c8aa4
fxmarty fxmarty requested a review from michaelbenayoun michaelbenayoun 2 years ago
fxmarty fxmarty requested a review from JingyaHuang JingyaHuang 2 years ago
HuggingFaceDocBuilderDev
regisss
regisss approved these changes on 2022-12-22
fxmarty handle device for list and tuple
9b0c5420
michaelbenayoun
michaelbenayoun approved these changes on 2022-12-22
fxmarty fix test
9ca6ab94
fxmarty Update optimum/exporters/onnx/convert.py
76e8a519
fxmarty Update optimum/exporters/onnx/convert.py
b17637c1
fxmarty fix test
3c58408b
fxmarty fix torch import
2c4d249f
fxmarty fix import2
83065c85
fxmarty debug
bbf58990
fxmarty run all
0e306d8a
fxmarty fxmarty merged 874d2b25 into main 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum