Link Search Menu Expand Document

Text Generation

Initialize a HappyTextGeneration() object to perform text generation

Initialization Arguments:

  1. model_type (string): Specify the model name in all caps, such as “ROBERTA” or “ALBERT”
  2. model_name(string): below is a URL that contains potential models: MODELS

We recommend using “HappyTextGeneration(“GPT2”, “gpt2-xl”)” for the best performance. If you are using Google Colab on a free instance, then we recommended using “gpt2-large”. “gpt2” is the default.

Example 1.0:

    from happytransformer import HappyGeneration
    # --------------------------------------#
    happy_gen = HappyGeneration("GPT2", "gpt2")  # default
    # happy_gen = HappyGeneration("GPT2", "gpt2-large")  # Good for Google Colab
    # happy_gen = HappyGeneration("GPT2", "gpt2-xl")  # Best performance 

Courses

Create a text generation web app. Also learn how to fine-tune GPT-Neo

Tutorials

Text generation with training (GPT-Neo)


Table of contents