-
Notifications
You must be signed in to change notification settings - Fork 25.9k
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
Add an example of exporting BartModel + BeamSearch to ONNX module. #13765
Conversation
Merge necessary updates from master.
Think we just need to run |
Thanks for adding this! Could you perhaps add a README to this example, to explain what the example is about and how to use it? |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks for this @fatcat-z.
Could you please send another PR that adds a README and addresses my comments?
Hey @fatcat-z, kindly pinging you for taking into account the comments on this PR (perhaps in a follow-up PR), if possible! |
I'm working on a new PR to address your and Gary's comments. |
…uggingface#13765) * Add all example files. * Reformat files by black. * Style. * Remove unused imports. Co-authored-by: Morgan Funtowicz <funtowiczmo@gmail.com>
This PR is aimed to deliver an example to describe how to export BartModel + BeamSearch to ONNX model and inference it by ONNXRuntime.
These new example files were added without updates to existing Huggingface model code. It mainly focuses on: