update download stuff

2019-03-04 10:36:44 -08:00
parent 8eb67930d7
commit ed0dedcd55
5 changed files with 29 additions and 51 deletions
--- a/Dockerfile.gpu
+++ b/Dockerfile.gpu
@ -12,10 +12,6 @@ ENV NVIDIA_VISIBLE_DEVICES=all \

 RUN mkdir /gpt-2
 WORKDIR /gpt-2
-COPY requirements.txt download_model.sh /gpt-2/
-RUN apt-get update && \
-    apt-get install -y curl && \
-    sh download_model.sh 117M
-RUN pip3 install -r requirements.txt
-
 ADD . /gpt-2
+RUN pip3 install -r requirements.txt
+RUN python3 download_model.py 117M