Exporters onnx module #403
michaelbenayoun
merged
39 commits into
huggingface:main
from
michaelbenayoun:exporters_onnx_module
[WIP] ONNX export in optimum => Bert works
77311770
[WIP] ONNX export in optimum => GPT-2 / GPT Neo
dd0ab63a
[WIP] ONNX export in optimum => T5
15e9b8cf
[WIP] ONNX export in optimum
0b896bf1
[WIP] Add support for TensorFlow export
2023c691
[WIP] onnx export
180d7ddb
Update file headers
e63a0308
[WIP] cache features.py
91dca2e2
Refactor features.py
a9678991
Bert like models ready
f0b2505b
Apply suggestions
1c889669
Clean up
e90cb180
regisss
approved these changes
on 2022-10-11
Backup input_generators.py version
b0bdd46a
Most seq2seq models work
2cab2676
Add vision models
c6f043f7
Support for GPT-J
25b8c5c6
CLIP is working
8bf6c80b
Add support for the remaining models
1a85ef99
Make style
918947f7
Fix workflow and remove commented code
9b8c3ee6
nit
3b81d6cb
Renamed dynamic axes
fd5654df
Fix BART
c01f5ec8
Drop support for PyTorch 1.10 and lower
fd0a302b
Add re-order inputs method
25c253f6
Small fixes
fbcbfb41
Changed FeaturesManager to TasksManager
0bb9af2f
Apply suggestions
0e8f19c9
regisss
commented
on 2022-10-27
Fix Bloom export
e5e0152a
Fix mt5 export
55f683ab
Change ATOL_FOR_VALIDATION for ResNet
89705c86
Update github actions for exporters to use pytest
40ad484f
Styling
96f2e0a2
Remove old comment
057958e7
Fix github action for exporters
97e9f086
Fix tests that are not passing
217a7f02
Styling
e1862d14
Change Blenderbot model to use
3ffb7e55
Change Resnet ATOL value
868e1a20
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub