move to azure

2020-12-02 12:56:19 -08:00
parent 0574c5708b
commit a74da5d99a
1 changed files with 1 additions and 1 deletions
--- a/download_model.py
+++ b/download_model.py
@ -16,7 +16,7 @@ subdir = subdir.replace('\\','/') # needed for Windows

 for filename in ['checkpoint','encoder.json','hparams.json','model.ckpt.data-00000-of-00001', 'model.ckpt.index', 'model.ckpt.meta', 'vocab.bpe']:

-    r = requests.get("https://storage.googleapis.com/gpt-2/" + subdir + "/" + filename, stream=True)
+    r = requests.get("https://openaipublic.blob.core.windows.net/gpt-2/" + subdir + "/" + filename, stream=True)

    with open(os.path.join(subdir, filename), 'wb') as f:
        file_size = int(r.headers["content-length"])