updates for 345M model

2019-05-02 20:39:33 -07:00
parent d14501aade
commit 0503b1b249
5 changed files with 9 additions and 5 deletions
--- a/Dockerfile.gpu
+++ b/Dockerfile.gpu
@ -15,3 +15,4 @@ WORKDIR /gpt-2
 ADD . /gpt-2
 RUN pip3 install -r requirements.txt
 RUN python3 download_model.py 117M
+RUN python3 download_model.py 345M