Skip to content

Commit

Permalink
Update ds-chat workflow to work w/ deepspeed-chat install (#4598)
Browse files Browse the repository at this point in the history
This PR updates the ds-chat workflow to work with the `deepspeed-chat`
install once this PR completes in DeepSpeedExamples:
microsoft/DeepSpeedExamples#731

Manual workflow run:

https://github.com/microsoft/DeepSpeed/actions/runs/6738638809/job/18318636818
  • Loading branch information
lekurile authored Nov 8, 2023
1 parent 217b15d commit 8ad50cf
Showing 1 changed file with 11 additions and 6 deletions.
17 changes: 11 additions & 6 deletions .github/workflows/nv-ds-chat.yml
Original file line number Diff line number Diff line change
Expand Up @@ -36,11 +36,7 @@ jobs:
pip install .[dev]
ds_report
- name: Python environment
run: |
pip list
- name: DS-Chat unit tests
- name: Install deepspeed-chat
run: |
BRANCH="master"
if [[ ! -z "${{ github.event.inputs.dse_branch }}" ]]; then
Expand All @@ -50,8 +46,17 @@ jobs:
git clone -b $BRANCH https://github.com/microsoft/DeepSpeedExamples.git
cd DeepSpeedExamples/applications/DeepSpeed-Chat
pip install -r requirements.txt
pip install -e .
- name: Python environment
run: |
pip list
- name: DS-Chat unit tests
run: |
cd DeepSpeedExamples/applications/DeepSpeed-Chat
unset TORCH_CUDA_ARCH_LIST # only jit compile for current arch
cd training/tests
cd tests
pytest $PYTEST_OPTS ./
- name: Open GitHub issue if nightly CI fails
Expand Down

0 comments on commit 8ad50cf

Please sign in to comment.