Skip to content

Update GPT-2 notebook with text generation example#4799

Merged
tianleiwu merged 1 commit into
masterfrom
tlwu/gpt2_notebook
Aug 17, 2020
Merged

Update GPT-2 notebook with text generation example#4799
tianleiwu merged 1 commit into
masterfrom
tlwu/gpt2_notebook

Conversation

@tianleiwu
Copy link
Copy Markdown
Contributor

Description:
Update GPT-2 notebook:
(1) Convert PyTorch model to ONNX model
(2) Add example of using IO Binding and past state for text generation.

Motivation and Context

  • Why is this change required? What problem does it solve?
  • If it fixes an open issue, please link to the issue here.

Example of using ONNX Runtime for GPT-2 text generation.

@tianleiwu tianleiwu requested a review from a team as a code owner August 14, 2020 08:06
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