mir/gpt-2
2
0
Fork 0
mirror of https://github.com/openai/gpt-2 synced 2025-08-22 18:19:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

move to azure

Browse Source
This commit is contained in:
Jeff Wu 2020-12-02 12:56:19 -08:00
parent 0574c5708b
commit a74da5d99a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
download_model.py
View File

@ -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']: 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: with open(os.path.join(subdir, filename), 'wb') as f:
file_size = int(r.headers["content-length"]) file_size = int(r.headers["content-length"])