Unverified Commit 11cb6e0f authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Unpin DeepSpeed and require DS >= 0.9.3 (#24541)



* fix

* fix

---------
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent e84bf1f7
...@@ -393,7 +393,7 @@ jobs: ...@@ -393,7 +393,7 @@ jobs:
working-directory: /workspace working-directory: /workspace
run: | run: |
python3 -m pip uninstall -y deepspeed 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 - name: NVIDIA-SMI
run: | run: |
......
...@@ -107,7 +107,7 @@ _deps = [ ...@@ -107,7 +107,7 @@ _deps = [
"dataclasses", "dataclasses",
"datasets!=2.5.0", "datasets!=2.5.0",
"decord==0.6.0", "decord==0.6.0",
"deepspeed>=0.8.3", "deepspeed>=0.9.3",
"diffusers", "diffusers",
"dill<0.3.5", "dill<0.3.5",
"evaluate>=0.2.0", "evaluate>=0.2.0",
......
...@@ -12,7 +12,7 @@ deps = { ...@@ -12,7 +12,7 @@ deps = {
"dataclasses": "dataclasses", "dataclasses": "dataclasses",
"datasets": "datasets!=2.5.0", "datasets": "datasets!=2.5.0",
"decord": "decord==0.6.0", "decord": "decord==0.6.0",
"deepspeed": "deepspeed>=0.8.3", "deepspeed": "deepspeed>=0.9.3",
"diffusers": "diffusers", "diffusers": "diffusers",
"dill": "dill<0.3.5", "dill": "dill<0.3.5",
"evaluate": "evaluate>=0.2.0", "evaluate": "evaluate>=0.2.0",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment