DEV Community

Cover image for Language Generation with OpenAI's GPT-2 in Python
James Briggs
James Briggs

Posted on

Language Generation with OpenAI's GPT-2 in Python

Easy natural language generation with Transformers and PyTorch. We apply OpenAI's GPT-2 model to generate text in just a few lines of Python code.

Language generation is one of those natural language tasks that can really produce an incredible feeling of awe at how far the fields of machine learning and artificial intelligence have come.

GPT-1, 2, and 3 are OpenAI’s top language models — well known for their ability to produce incredibly natural, coherent, and genuinely interesting language.

In this article, we will take a small snippet of text and learn how to feed that into a pre-trained GPT-2 model using PyTorch and Transformers to produce high-quality language generation in just eight lines of code. We cover:

PyTorch and Transformers

  • Data Building the Model
  • Initialization
  • Tokenization
  • Generation
  • Decoding Results

Top comments (0)