Pin nv-a6000 workflow (#6938)

Breaking change in transformers is huggingface/transformers#35235. Need to make changes to unpin nv-a6000 workflow.
microsoft · Jan 13, 2025 · 66d3d3e · 66d3d3e
1 parent 396f8db
commit 66d3d3e
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.1
           git rev-parse --short HEAD
           python -m pip install .
       - name: Install deepspeed