gpt-2

Code for the paper &#34;Language Models are Unsupervised Multitask Learners&#34;

Code and samples from the paper "Language Models are Unsupervised Multitask Learners"

Installation

Download the model data:

gsutil rsync -r gs://gpt-2/models/ models/

Install python packages:

pip install -r requirements.txt

WARNING: Samples are unfiltered and may contain offensive content.

To generate unconditional samples from the small model:

python3 src/main.py | tee samples

There are various flags for controlling the samples:

python3 src/main.py --top_k 40 --temperature 0.7 | tee samples