pin pydantic so deepspeed isn't broken

2023-07-02 22:26:51 -04:00
parent 3a783c04e4
commit 71456955f5
1 changed files with 1 additions and 1 deletions
--- a/docker/Dockerfile-base
+++ b/docker/Dockerfile-base
@@ -97,4 +97,4 @@ RUN cd /workspace/builds/bitsandbytes && python3 setup.py install
 RUN git lfs install --skip-repo
 RUN pip3 install awscli && \
    # The base image ships with `pydantic==1.8.2` which is not working
-    pip3 install -U --no-cache-dir pydantic
+    pip3 install -U --no-cache-dir pydantic==1.10.10