4 Commits

Author	SHA1	Message	Date
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