All Projects → SimiaCryptus → tf-gpt-2

SimiaCryptus / tf-gpt-2

Licence: Apache-2.0 license
Java library for the GPT-2 Text Model using Tensorflow

Programming Languages

java
68154 projects - #9 most used programming language

tf-gpt-2

Java library for the GPT-2 Text Model using Tensorflow

Source:

  1. Original Python Release
  2. OpenAI Blog - GPT2 Details

More Background:

  1. Transformers and Attention Models

Basic Usage

Import the Library

<dependency>
    <groupId>com.simiacryptus</groupId>
    <artifactId>tf-gpt-2</artifactId>
    <version>1.7.1</version>
</dependency>

Instantiate the text generator

import com.simiacryptus.text.TextGenerator;
import com.simiacryptus.text.gpt2.GPT2Util;
TextGenerator textGenerator = GPT2Util.get345M();

Generate text

System.out.println(textGenerator.generateText(500));

Generate text given prefix

System.out.println(textGenerator.generateText(500, "Once upon a time"));
Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].