From 89f4bc162c93914c7e4a8e43f6d84e9824fcab45 Mon Sep 17 00:00:00 2001
From: Jeff Wu <wuthefwasthat@gmail.com>
Date: Thu, 14 Feb 2019 09:12:05 -0800
Subject: [PATCH] fix downloading

---
 README.md         | 4 ++--
 download_model.sh | 8 ++++++++
 2 files changed, 10 insertions(+), 2 deletions(-)
 create mode 100644 download_model.sh

diff --git a/README.md b/README.md
index c853848..9df579e 100644
--- a/README.md
+++ b/README.md
@@ -10,12 +10,12 @@ See more details in our [blog post](https://blog.openai.com/better-language-mode
 
 Download the model data (needs [gsutil](https://cloud.google.com/storage/docs/gsutil_install)):
 ```
-mkdir models && gsutil rsync -r gs://gpt-2/models/ models/
+sh download_model.sh 117M
 ```
 
 Install python packages:
 ```
-pip install -r requirements.txt
+pip3 install -r requirements.txt
 ```
 
 ## Sample generation
diff --git a/download_model.sh b/download_model.sh
new file mode 100644
index 0000000..f420442
--- /dev/null
+++ b/download_model.sh
@@ -0,0 +1,8 @@
+model=$1
+
+mkdir models/$model
+
+# TODO: gsutil rsync -r gs://gpt-2/models/ models/
+for filename in checkpoint encoder.json hparams.json model.ckpt.data-00000-of-00001 model.ckpt.index model.ckpt.meta vocab.bpe; do
+  gsutil cp gs://gpt-2/models/$model/$filename models/$model/
+done