Pin transformers on a6000 tests

deepspeedai · Jan 9, 2025 · cd228d3 · cd228d3
1 parent 53fb579
commit cd228d3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.github/workflows/nv-a6000.yml b/.github/workflows/nv-a6000.yml
@@ -40,8 +40,9 @@ jobs:
           python -c "import torch; print('CUDA available:', torch.cuda.is_available())"
       - name: Install transformers
         run: |
-          git clone --depth=1 https://github.com/huggingface/transformers
+          git clone https://github.com/huggingface/transformers
           cd transformers
+          git checkout v4.47.0
           git rev-parse --short HEAD
           python -m pip install .
       - name: Install deepspeed