Skip to content

Commit

Permalink
Update deepspeed.yaml
Browse files Browse the repository at this point in the history
  • Loading branch information
xuedinge233 authored Oct 22, 2024
1 parent 6593f87 commit 1c81cd9
Showing 1 changed file with 10 additions and 10 deletions.
20 changes: 10 additions & 10 deletions .github/workflows/deepspeed.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -69,23 +69,23 @@ jobs:
print(f"Device Available: {torch.npu.is_available()}")
EOF
- name: Install transformers
run: |
source /root/.bashrc
echo "y" | apt-get install git
git clone https://github.com/huggingface/transformers
cd transformers
git rev-parse --short HEAD
pip install . -i https://pypi.tuna.tsinghua.edu.cn/simple
# - name: Install transformers
# run: |
# source /root/.bashrc
# echo "y" | apt-get install git
# git clone https://github.com/huggingface/transformers
# cd transformers
# git rev-parse --short HEAD
# pip install .

- name: Install deepspeed
run: |
source /root/.bashrc
echo "y" | apt-get install git
git clone --depth=1 https://github.com/microsoft/DeepSpeed.git
cd DeepSpeed
pip install .[1bit,autotuning,inf]
pip install -r ../requirements/requirements_deepspeed.txt
pip install -r ../requirements/requirements_deepspeed.txt
pip install .[1bit,autotuning,inf]
ds_report
- name: Python environment
Expand Down

0 comments on commit 1c81cd9

Please sign in to comment.