The package wraps OpenAI’s GPT-2 model management and Neil Sheppard’s finetuning code with simplified Python APIs. Users can retrain GPT-2 on their own texts using a GPU or generate text from a fine-tuned model, optionally with text prefixes to control output. Checkpoints trained with this package can be loaded using its successor aitextgen.