onnxruntime/onnxruntime/python/tools
kunal-vaishnavi 18e81f8785
Fix Whisper export for FP16 CUDA (#22410)
### Description
This PR fixes a bug when the ONNX checker is called while exporting
Whisper for FP16 CUDA with optional flags.

### Motivation and Context
Sometimes, the ONNX checker raises an error depending on the optional
flags passed. By wrapping the ONNX checker in a try-except, the
conversion can continue even if the checker fails.
2024-10-11 17:37:36 -07:00
..
custom_op_wrapper
kernel_explorer hipify int4 gemv (#20666) 2024-05-18 16:59:03 +08:00
microbench Bump ruff to 0.3.2 and black to 24 (#19878) 2024-03-13 10:00:32 -07:00
profile_explorer [Linter] Bump ruff and remove pylint (#17797) 2023-10-05 21:07:33 -07:00
qnn Improve the script to add Q, DQ nodes around EPContext node (#20107) 2024-04-05 10:12:01 -07:00
quantization Add TensorRT-Model-Optimizer INT4 AWQ support in onnxruntime tools (#22390) 2024-10-11 13:31:54 -07:00
tensorrt [Running CI] Update TensorRT to 10.4 (#22049) 2024-09-26 11:10:52 -07:00
transformers Fix Whisper export for FP16 CUDA (#22410) 2024-10-11 17:37:36 -07:00
__init__.py
offline_tuning.py
onnx_randomizer.py
onnxruntime_test.py [Linter] Bump ruff and remove pylint (#17797) 2023-10-05 21:07:33 -07:00
pytorch_export_contrib_ops.py Update ruff and clang-format versions (#21479) 2024-07-24 11:50:11 -07:00
symbolic_shape_infer.py Update ruff and clang-format versions (#21479) 2024-07-24 11:50:11 -07:00