Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

T5 with past ONNX export #13014

Merged
merged 3 commits into from
Aug 6, 2021
Merged

Conversation

michaelbenayoun
Copy link
Member

This PR enables the export of T5 with past keys and values to ONNX.

It also enhances the ONNX export when using past keys and values by making the inputs and outputs names for past_key_values more explicit and easy to understand.

Copy link
Collaborator

@sgugger sgugger left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for fixing this!

@michaelbenayoun michaelbenayoun marked this pull request as ready for review August 6, 2021 08:55
@michaelbenayoun michaelbenayoun merged commit dc420b0 into huggingface:master Aug 6, 2021
LysandreJik pushed a commit that referenced this pull request Aug 9, 2021
T5 with past ONNX export, and more explicit past_key_values inputs and outputs names for ONNX model

Authored-by: Michael Benayoun <michael@huggingface.co>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants