tests/onnx/test_onnx_v2.py · 0d4c45c585fadd0d9339061feda0f22fce04c57d · chenpangpang / transformers

Add Onnx Config for ImageGPT (#19868) · 0d4c45c5

Raghav Prabhakar authored Oct 28, 2022



* add Onnx Config for ImageGPT

* add generate_dummy_inputs for onnx config

* add TYPE_CHECKING clause

* Update doc for generate_dummy_inputs
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>

0d4c45c5

test_onnx_v2.py 21.6 KB

Replace test_onnx_v2.py