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

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

View File

@@ -22,7 +22,7 @@ RUN apt update && \
apt clean && \
rm -rf /var/lib/apt/lists/*
RUN python3 -m pip install --no-cache-dir --upgrade pip ninja "pydantic<2"
RUN python3 -m pip install --no-cache-dir --upgrade pip ninja "pydantic>=2.0.0"
RUN python3 -m pip uninstall -y apex torch torchvision torchaudio
RUN python3 -m pip install torch==$PYTORCH torchvision==$TORCH_VISION torchaudio==$TORCH_AUDIO --index-url https://download.pytorch.org/whl/rocm$ROCM --no-cache-dir