Jingya HUANG
6da76b9c2a
Add onnx export cuda support ( #17183 )
...
Co-authored-by: Lysandre Debut <lysandre@huggingface.co>
Co-authored-by: lewtun <lewis.c.tunstall@gmail.com>
2022-05-18 17:52:13 +02:00
Sylvain Gugger
afe5d42d8d
Black preview ( #17217 )
...
* Black preview
* Fixup too!
* Fix check copies
* Use the same version as the CI
* Bump black
2022-05-12 16:25:55 -04:00
Manan Dey
dc3645dc9c
add mobilebert onnx configs ( #17029 )
...
* update docs of length_penalty
* Revert "update docs of length_penalty"
This reverts commit 466bf4800b
2022-05-09 10:36:53 -04:00
Ritik Nandwal
215e0681e4
Added BigBirdPegasus onnx config ( #17104 )
...
* Add onnx configuration for bigbird-pegasus
* Modify docs
2022-05-06 17:31:00 +02:00
lewtun
4bb1d0ec84
Skip RoFormer ONNX test if rjieba not installed ( #16981 )
...
* Skip RoFormer ONNX test if rjieba not installed
* Update deps table
* Skip RoFormer serialization test
* Fix RoFormer vocab
* Add rjieba to CircleCI
2022-05-04 10:04:10 +02:00
Krishna Sirumalla
aaee4038c3
Add onnx config for RoFormer ( #16861 )
...
* add roformer onnx config
2022-04-26 16:51:15 +02:00
Rushi Chaudhari
8246caf3eb
added deit onnx config ( #16887 )
...
* added deit onnx config
2022-04-25 20:50:45 +02:00
Thomas Chaigneau
508baf1943
add bigbird typo fixes ( #16897 )
...
Co-authored-by: ChainYo <t.chaigneau.tc@gmail.com>
2022-04-25 11:32:06 +02:00
Thomas Chaigneau
ec81c11a18
Add OnnxConfig for ConvBERT ( #16859 )
...
* add OnnxConfig for ConvBert
Co-authored-by: ChainYo <t.chaigneau.tc@gmail.com>
2022-04-22 18:19:15 +02:00
Ella Charlaix
77de8d6c31
Add onnx export of models with a multiple choice classification head ( #16758 )
...
* Add export of models with a multiple-choice classification head
2022-04-19 15:51:51 +02:00
Minh Chien Vu
9c9db751e2
add Bigbird ONNX config ( #16427 )
...
* add Bigbird ONNX config
2022-04-12 20:46:06 +02:00
Jim Rohrer
9de70f213e
Add ONNX export for BeiT ( #16498 )
...
* Add beit onnx conversion support
* Updated docs
* Added cross reference to ViT ONNX config
2022-04-01 10:52:42 +02:00
lewtun
a97f3150c4
Add ONNX support for Blenderbot and BlenderbotSmall ( #15875 )
...
* Add ONNX support for Blenderbot
* Add BlenderbotSmall ONNX configuration
* Update serialization table
2022-03-25 17:04:43 +01:00
Sylvain Gugger
4975002df5
Reorganize file utils ( #16264 )
...
* Split file_utils in several submodules
* Fixes
* Add back more objects
* More fixes
* Who exactly decided to import that from there?
* Second suggestion to code with code review
* Revert wront move
* Fix imports
* Adapt all imports
* Adapt all imports everywhere
* Revert this import, will fix in a separate commit
2022-03-23 10:26:33 -04:00
lewtun
6e1e88fd38
Add TFCamembertForCausalLM and ONNX integration test ( #16073 )
...
* Make Camembert great again!
* Add Camembert to TensorFlow ONNX tests
2022-03-14 08:40:42 +01:00
lewtun
6b09328368
Fix duplicate arguments passed to dummy inputs in ONNX export ( #16045 )
...
* Fix duplicate arguments passed to dummy inputs in ONNX export
* Fix M2M100 ONNX config
* Ensure we check PreTrained model only if torch is available
* Remove TensorFlow tests for models without PyTorch parity
2022-03-10 20:19:45 +01:00
lewtun
50dd314d93
Add ONNX export for ViT ( #15658 )
...
* Add ONNX support for ViT
* Refactor to use generic preprocessor
* Add vision dep to tests
* Extend ONNX slow tests to ViT
* Add dummy image generator
* Use model_type to determine modality
* Add deprecation warnings for tokenizer argument
* Add warning when overwriting the preprocessor
* Add optional args to docstrings
* Add minimum PyTorch version to OnnxConfig
* Refactor OnnxConfig class variables from CONSTANT_NAME to snake_case
* Add reasonable value for default atol
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
2022-03-09 17:36:59 +01:00
Michael Benayoun
4bfe75bd08
M2M100 support for ONNX export ( #15193 )
...
* Add M2M100 support for ONNX export
* Delete useless imports
* Add M2M100 to tests
* Fix protobuf issue
2022-03-02 10:03:14 +01:00
Lysandre Debut
29c10a41d0
[Test refactor 1/5] Per-folder tests reorganization ( #15725 )
...
* Per-folder tests reorganization
Co-authored-by: sgugger <sylvain.gugger@gmail.com>
Co-authored-by: Stas Bekman <stas@stason.org>
2022-02-23 15:46:28 -05:00
