tests/onnx/test_onnx_v2.py · 4f38808e9e1b7ff9c7b783444c8df8493e44dcda · chenpangpang / transformers

Add OnnxConfig for SqueezeBert iss17314 (#17315) · 4f38808e

Ruihua Fang authored Jun 01, 2022



* add onnx config for SqueezeBert

* add test for onnx config for SqueezeBert

* add automatically updated doc for onnx config for SqueezeBert

* Update src/transformers/onnx/features.py
Co-authored-by: lewtun <lewis.c.tunstall@gmail.com>

* Update src/transformers/models/squeezebert/configuration_squeezebert.py
Co-authored-by: lewtun <lewis.c.tunstall@gmail.com>
Co-authored-by: lewtun <lewis.c.tunstall@gmail.com>

4f38808e

test_onnx_v2.py 14.4 KB

Replace test_onnx_v2.py