From e419a40cf45667a24738bd70bfb25eac742d2e99 Mon Sep 17 00:00:00 2001 From: Logan Adams Date: Thu, 5 Dec 2024 11:00:37 -0800 Subject: [PATCH] Pin transformers version - latest 4.47 release breaks CI due to changes in processing/threading --- .github/workflows/cpu-torch-latest.yml | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/.github/workflows/cpu-torch-latest.yml b/.github/workflows/cpu-torch-latest.yml index 0125fa50bc14..51bc60c2c2ae 100644 --- a/.github/workflows/cpu-torch-latest.yml +++ b/.github/workflows/cpu-torch-latest.yml @@ -37,6 +37,15 @@ jobs: python -c "import torch; print('torch:', torch.__version__, torch)" python -c "import torch; print('CUDA available:', torch.cuda.is_available())" + - name: Install transformers + run: | + git clone https://github.com/huggingface/transformers + cd transformers + # if needed switch to the last known good SHA until transformers@master is fixed + git checkout 6c3f168b3 + git rev-parse --short HEAD + pip install . + - name: Install deepspeed run: | pip install .[dev,autotuning]