From 8ad50cf089d628f36da1095786193acdbeb566c9 Mon Sep 17 00:00:00 2001 From: Lev Kurilenko <113481193+lekurile@users.noreply.github.com> Date: Wed, 8 Nov 2023 13:24:35 -0800 Subject: [PATCH] Update ds-chat workflow to work w/ deepspeed-chat install (#4598) This PR updates the ds-chat workflow to work with the `deepspeed-chat` install once this PR completes in DeepSpeedExamples: https://github.com/microsoft/DeepSpeedExamples/pull/731 Manual workflow run: https://github.com/microsoft/DeepSpeed/actions/runs/6738638809/job/18318636818 --- .github/workflows/nv-ds-chat.yml | 17 +++++++++++------ 1 file changed, 11 insertions(+), 6 deletions(-) diff --git a/.github/workflows/nv-ds-chat.yml b/.github/workflows/nv-ds-chat.yml index c90b7c72ac0b..b53fac36315b 100644 --- a/.github/workflows/nv-ds-chat.yml +++ b/.github/workflows/nv-ds-chat.yml @@ -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 @@ -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