From c4745e6949cce4b93d8f0fdfa3a50bb2eb948824 Mon Sep 17 00:00:00 2001 From: Simon Willison Date: Fri, 31 Jan 2020 16:44:48 -0800 Subject: [PATCH] tensorflow==1.15.2 Tensorflow 2 is incompatible with this script. 1.15.2 is the most recent release of Tensorflow 1. --- Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index 23adf76..22fa24b 100644 --- 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.