docker layer caching, build w axolotl from base build

2023-05-07 17:16:05 -04:00
parent 0c46806ae2
commit 990bec63e6
6 changed files with 27 additions and 19 deletions
--- a/docker/Dockerfile-base
+++ b/docker/Dockerfile-base
@@ -66,4 +66,7 @@ RUN git lfs install --skip-repo
 RUN pip3 install "peft @ git+https://github.com/huggingface/peft.git@main" \
        "accelerate @ git+https://github.com/huggingface/accelerate.git@main" \
        "transformers @ git+https://github.com/huggingface/transformers.git@main" && \
-    pip3 install awscli
+    pip3 install awscli && \
+    # The base image ships with `pydantic==1.8.2` which is not working
+    pip3 install -U --no-cache-dir pydantic
+