fix deepspeed job (#37284)

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
Yih-Dar 2025-04-08 15:19:33 +02:00 committed by GitHub
parent cdfb018d03
commit 8064cd9b4f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -336,6 +336,10 @@ jobs:
working-directory: ${{ inputs.working-directory-prefix }}/transformers working-directory: ${{ inputs.working-directory-prefix }}/transformers
run: git fetch && git checkout ${{ github.sha }} 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) - name: Reinstall transformers in edit mode (remove the one installed during docker image build)
working-directory: ${{ inputs.working-directory-prefix }}/transformers working-directory: ${{ inputs.working-directory-prefix }}/transformers
run: python3 -m pip uninstall -y transformers && python3 -m pip install -e . run: python3 -m pip uninstall -y transformers && python3 -m pip install -e .
@ -382,7 +386,7 @@ jobs:
run: pip freeze run: pip freeze
- name: Set `machine_type` for report and artifact names - name: Set `machine_type` for report and artifact names
working-directory: /transformers working-directory: ${{ inputs.working-directory-prefix }}/transformers
shell: bash shell: bash
run: | run: |
echo "${{ matrix.machine_type }}" echo "${{ matrix.machine_type }}"