SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

fix: Fixed pydantic required version in dockerfiles to make it compatible with DeepSpeed (#33105)

Browse Source 
Fixed pydantic required version in dockerfiles.
This commit is contained in:
Sai-Suraj-27
2024-08-26 20:40:36 +05:30
committed by GitHub
parent a378a54a57
commit 3562772969
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docker/transformers-pytorch-deepspeed-latest-gpu/Dockerfile
View File

@@ -49,5 +49,5 @@ RUN DS_BUILD_CPU_ADAM=1 DS_BUILD_FUSED_ADAM=1 python3 -m pip install deepspeed -
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
RUN python3 -m pip install -U --no-cache-dir "pydantic<2"
RUN python3 -m pip install -U --no-cache-dir "pydantic>=2.0.0"
RUN python3 -c "from deepspeed.launcher.runner import main"