fix pydantic install command
This commit is contained in:
@@ -50,5 +50,5 @@ RUN DS_BUILD_CPU_ADAM=1 DS_BUILD_FUSED_ADAM=1 DS_BUILD_UTILS=1 python3 -m pip in
|
|||||||
RUN cd transformers && python3 setup.py develop
|
RUN cd transformers && python3 setup.py develop
|
||||||
|
|
||||||
# The base image ships with `pydantic==1.8.2` which is not working - i.e. the next command fails
|
# The base image ships with `pydantic==1.8.2` which is not working - i.e. the next command fails
|
||||||
RUN python3 -m pip install -U --no-cache-dir pydantic<2
|
RUN python3 -m pip install -U --no-cache-dir "pydantic<2"
|
||||||
RUN python3 -c "from deepspeed.launcher.runner import main"
|
RUN python3 -c "from deepspeed.launcher.runner import main"
|
||||||
|
|||||||
Reference in New Issue
Block a user