Merge pull request minimaxir#19 from simonw/patch-1

tensorflow==1.15.2
Hugo0 · Feb 1, 2020 · 6f118ff · 6f118ff
2 parents ef992f4 + c4745e6
commit 6f118ff
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -8,7 +8,7 @@ COPY checkpoint /checkpoint
 # Make changes to the requirements/app here.
 # This Dockerfile order allows Docker to cache the checkpoint layer
 # and improve build times if making changes.
-RUN pip3 --no-cache-dir install tensorflow gpt-2-simple starlette uvicorn ujson
+RUN pip3 --no-cache-dir install tensorflow==1.15.2 gpt-2-simple starlette uvicorn ujson
 COPY app.py /
 
 # Clean up APT when done.