From ac100ebb649a723d65bc73bb19051cb375a4f000 Mon Sep 17 00:00:00 2001 From: pengwa Date: Mon, 3 Jul 2023 13:18:59 +0800 Subject: [PATCH] Fix orttraining-ortmodule-distributed CI (#16569) ### Fix orttraining-ortmodule-distributed CI https://pypi.org/project/pydantic/#history released version 2.0 1st July, Deepspeed has known issue on newer version of it (https://github.com/microsoft/DeepSpeed/issues/3280). So fix this by add similar check as DS did in https://github.com/microsoft/DeepSpeed/pull/3290 --- .../docker/scripts/training/ortmodule/stage2/requirements.txt | 1 + 1 file changed, 1 insertion(+) diff --git a/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage2/requirements.txt b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage2/requirements.txt index d1d4f1c65dc9..202d43befcca 100644 --- a/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage2/requirements.txt +++ b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage2/requirements.txt @@ -11,3 +11,4 @@ pytorch-lightning deepspeed==0.9.0 fairscale==0.4.6 parameterized>=0.8.1 +pydantic<2.0.0