From cd228d3b03e4feb94c185f38e21b7e11d265b241 Mon Sep 17 00:00:00 2001 From: Logan Adams Date: Thu, 9 Jan 2025 08:36:55 -0800 Subject: [PATCH] Pin transformers on a6000 tests --- .github/workflows/nv-a6000.yml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/.github/workflows/nv-a6000.yml b/.github/workflows/nv-a6000.yml index 639f27498dd9..0eee9d057d03 100644 --- 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