Undo pinning transformers

microsoft · Sep 3, 2024 · 6fc3986 · 6fc3986
1 parent 805a9bc
commit 6fc3986
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/.github/workflows/nv-a6000-fastgen.yml b/.github/workflows/nv-a6000-fastgen.yml
@@ -35,9 +35,8 @@ jobs:
           python -c "import torch; print('CUDA available:', torch.cuda.is_available())"
       - name: Install transformers
         run: |
-          git clone https://github.com/huggingface/transformers
+          git clone --depth=1 https://github.com/huggingface/transformers
           cd transformers
-          git checkout v4.44.2
           git rev-parse --short HEAD
           python -m pip install .
       - name: Install deepspeed