transformers
055afdb6 - Fix ONNX export for sequence classification head (#36332)

Commit

1 year ago

Fix ONNX export for sequence classification head (#36332) * set dtype to int32 * fix style

References

Author

echarlaix

Parents