diff --git a/.github/workflows/self-scheduled.yml b/.github/workflows/self-scheduled.yml index d92fb01cd..a0a9d3a5d 100644 --- a/.github/workflows/self-scheduled.yml +++ b/.github/workflows/self-scheduled.yml @@ -393,7 +393,7 @@ jobs: working-directory: /workspace run: | python3 -m pip uninstall -y deepspeed - DS_BUILD_CPU_ADAM=1 DS_BUILD_FUSED_ADAM=1 DS_BUILD_UTILS=1 python3 -m pip install deepspeed==0.9.2 --global-option="build_ext" --global-option="-j8" --no-cache -v --disable-pip-version-check + DS_BUILD_CPU_ADAM=1 DS_BUILD_FUSED_ADAM=1 DS_BUILD_UTILS=1 python3 -m pip install deepspeed --global-option="build_ext" --global-option="-j8" --no-cache -v --disable-pip-version-check - name: NVIDIA-SMI run: | diff --git a/setup.py b/setup.py index b88c7efe8..17a3e8c12 100644 --- a/setup.py +++ b/setup.py @@ -107,7 +107,7 @@ _deps = [ "dataclasses", "datasets!=2.5.0", "decord==0.6.0", - "deepspeed>=0.8.3", + "deepspeed>=0.9.3", "diffusers", "dill<0.3.5", "evaluate>=0.2.0", diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py index 1e585889c..9c525d791 100644 --- a/src/transformers/dependency_versions_table.py +++ b/src/transformers/dependency_versions_table.py @@ -12,7 +12,7 @@ deps = { "dataclasses": "dataclasses", "datasets": "datasets!=2.5.0", "decord": "decord==0.6.0", - "deepspeed": "deepspeed>=0.8.3", + "deepspeed": "deepspeed>=0.9.3", "diffusers": "diffusers", "dill": "dill<0.3.5", "evaluate": "evaluate>=0.2.0",