Skip to content

Commit

Permalink
Pin deepspeed, not transformers
Browse files Browse the repository at this point in the history
  • Loading branch information
loadams committed Jan 9, 2025
1 parent 4b50e4c commit bcb86a7
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion .github/workflows/nv-a6000.yml
Original file line number Diff line number Diff line change
Expand Up @@ -42,12 +42,13 @@ jobs:
run: |
git clone https://github.com/huggingface/transformers
cd transformers
git checkout v4.47.0
#git checkout v4.47.0
git rev-parse --short HEAD
python -m pip install .
- name: Install deepspeed
run: |
python -m pip install docutils==0.18.1 jinja2==3.0 urllib3==1.26.11 ninja
git checkout v0.15.2
python -m pip install .[dev,1bit,autotuning,inf]
ds_report
- name: Python environment
Expand Down

0 comments on commit bcb86a7

Please sign in to comment.