diff --git a/DEVELOPERS.md b/DEVELOPERS.md
index 57fd519..0992c70 100644
--- a/DEVELOPERS.md
+++ b/DEVELOPERS.md
@@ -27,8 +27,9 @@ pip3 install -r requirements.txt
Download the model data
```
-python3 download_model.py 117M
-python3 download_model.py 345M
+python3 download_model.py 124M
+python3 download_model.py 355M
+python3 download_model.py 774M
```
## Docker Installation
diff --git a/Dockerfile.cpu b/Dockerfile.cpu
index a02d2b3..c923234 100644
--- a/Dockerfile.cpu
+++ b/Dockerfile.cpu
@@ -5,5 +5,6 @@ RUN mkdir /gpt-2
WORKDIR /gpt-2
ADD . /gpt-2
RUN pip3 install -r requirements.txt
-RUN python3 download_model.py 117M
-RUN python3 download_model.py 345M
+RUN python3 download_model.py 124M
+RUN python3 download_model.py 355M
+RUN python3 download_model.py 774M
diff --git a/Dockerfile.gpu b/Dockerfile.gpu
index b3f87db..e59880e 100644
--- a/Dockerfile.gpu
+++ b/Dockerfile.gpu
@@ -14,5 +14,6 @@ RUN mkdir /gpt-2
WORKDIR /gpt-2
ADD . /gpt-2
RUN pip3 install -r requirements.txt
-RUN python3 download_model.py 117M
-RUN python3 download_model.py 345M
+RUN python3 download_model.py 124M
+RUN python3 download_model.py 355M
+RUN python3 download_model.py 774M
diff --git a/README.md b/README.md
index 46bb8f3..1b2d5e8 100644
--- a/README.md
+++ b/README.md
@@ -4,9 +4,11 @@
Code from the paper ["Language Models are Unsupervised Multitask Learners"](https://d4mucfpksywv.cloudfront.net/better-language-models/language-models.pdf).
-We have currently released small (117M parameter) and medium (345M parameter) versions of GPT-2. While we have not released the larger models, we have [released a dataset](https://github.com/openai/gpt-2-output-dataset) for researchers to study their behaviors.
+We have currently released small (124M parameter), medium (355M parameter), and large (774M parameter) versions of GPT-2*, with only the full model as of yet unreleased. We have also [released a dataset](https://github.com/openai/gpt-2-output-dataset) for researchers to study their behaviors.
-See more details in our [blog post](https://blog.openai.com/better-language-models/).
+You can read about GPT-2 and release decisions in our [original blog post](https://blog.openai.com/better-language-models/) and [6 month follow-up post](https://openai.com/blog/gpt-2-6-month-follow-up/).
+
+* *Note that our original parameter counts were wrong due to an error (in our previous blog posts and paper). Thus you may have seen small referred to as 117M and medium referred to as 345M.*
## Usage
diff --git a/download_model.py b/download_model.py
index 30ba84a..56d4e76 100644
--- a/download_model.py
+++ b/download_model.py
@@ -4,7 +4,7 @@ import requests
from tqdm import tqdm
if len(sys.argv) != 2:
- print('You must enter the model name as a parameter, e.g.: download_model.py 117M')
+ print('You must enter the model name as a parameter, e.g.: download_model.py 124M')
sys.exit(1)
model = sys.argv[1]
diff --git a/src/generate_unconditional_samples.py b/src/generate_unconditional_samples.py
index f18a838..cc3f3a3 100755
--- a/src/generate_unconditional_samples.py
+++ b/src/generate_unconditional_samples.py
@@ -9,7 +9,7 @@ import tensorflow as tf
import model, sample, encoder
def sample_model(
- model_name='117M',
+ model_name='124M',
seed=None,
nsamples=0,
batch_size=1,
@@ -20,7 +20,7 @@ def sample_model(
):
"""
Run the sample_model
- :model_name=117M : String, which model to use
+ :model_name=124M : String, which model to use
:seed=None : Integer seed for random number generators, fix seed to
reproduce results
:nsamples=0 : Number of samples to return, if 0, continues to
diff --git a/src/interactive_conditional_samples.py b/src/interactive_conditional_samples.py
index ae348d8..48b5cb3 100755
--- a/src/interactive_conditional_samples.py
+++ b/src/interactive_conditional_samples.py
@@ -9,18 +9,18 @@ import tensorflow as tf
import model, sample, encoder
def interact_model(
- model_name='117M',
+ model_name='124M',
seed=None,
nsamples=1,
batch_size=1,
length=None,
temperature=1,
top_k=0,
- models_dir='models',
+ models_dir='models',
):
"""
Interactively run the model
- :model_name=117M : String, which model to use
+ :model_name=124M : String, which model to use
:seed=None : Integer seed for random number generators, fix seed to reproduce
results
:nsamples=1 : Number of samples to return total
@@ -36,7 +36,7 @@ def interact_model(
while 40 means 40 words are considered at each step. 0 (default) is a
special setting meaning no restrictions. 40 generally is a good value.
:models_dir : path to parent folder containing model subfolders
- (i.e. contains the folder)
+ (i.e. contains the folder)
"""
models_dir = os.path.expanduser(os.path.expandvars(models_dir))
if batch_size is None: