update download stuff

This commit is contained in:
Jeff Wu
2019-03-04 10:36:44 -08:00
parent 8eb67930d7
commit ed0dedcd55
5 changed files with 29 additions and 51 deletions

View File

@ -17,12 +17,7 @@ Then, follow instructions for either native or Docker installation.
### Native Installation
Download the model data
```
sh download_model.sh 117M
```
The remaining steps can optionally be done in a virtual environment using tools such as `virtualenv` or `conda`.
All steps can optionally be done in a virtual environment using tools such as `virtualenv` or `conda`.
Install tensorflow 1.12 (with GPU support, if you have a GPU and want everything to run faster)
```
@ -38,6 +33,11 @@ Install other python packages:
pip3 install -r requirements.txt
```
Download the model data
```
python3 download_model.py 117M
```
### Docker Installation
Build the Dockerfile and tag the created image as `gpt-2`: