From 8064cd9b4fdd797843c2d884c309a3f2a35a5b25 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Tue, 8 Apr 2025 15:19:33 +0200 Subject: [PATCH] fix deepspeed job (#37284) Co-authored-by: ydshieh --- .github/workflows/self-scheduled.yml | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/.github/workflows/self-scheduled.yml b/.github/workflows/self-scheduled.yml index 78971820d1..dead87b5b6 100644 --- a/.github/workflows/self-scheduled.yml +++ b/.github/workflows/self-scheduled.yml @@ -336,6 +336,10 @@ jobs: working-directory: ${{ inputs.working-directory-prefix }}/transformers run: git fetch && git checkout ${{ github.sha }} + # TODO: update the docker image instead + - name: Reinstall some packages with specific versions + run: python3 -m pip install numpy==1.24.3 numba==0.61.0 scipy==1.12.0 scikit-learn==1.6.1 + - name: Reinstall transformers in edit mode (remove the one installed during docker image build) working-directory: ${{ inputs.working-directory-prefix }}/transformers run: python3 -m pip uninstall -y transformers && python3 -m pip install -e . @@ -382,7 +386,7 @@ jobs: run: pip freeze - name: Set `machine_type` for report and artifact names - working-directory: /transformers + working-directory: ${{ inputs.working-directory-prefix }}/transformers shell: bash run: | echo "${{ matrix.machine_type }}"