Skip to content

Latest commit

 

History

History
29 lines (22 loc) · 580 Bytes

README.md

File metadata and controls

29 lines (22 loc) · 580 Bytes

gpt-2

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

Installation

Download the model data:

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

Install python packages:

pip install -r requirements.txt

Sample generation

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