diff --git a/.github/workflows/deepspeed.yaml b/.github/workflows/deepspeed.yaml index a682d4974..befd410ac 100644 --- a/.github/workflows/deepspeed.yaml +++ b/.github/workflows/deepspeed.yaml @@ -87,6 +87,7 @@ jobs: cd DeepSpeed pip install -r ../requirements/requirements_deepspeed.txt pip install .[1bit,autotuning,inf] + pip install --upgrade deepspeed==0.15.2 pip list ds_report