From cd228d3b03e4feb94c185f38e21b7e11d265b241 Mon Sep 17 00:00:00 2001 From: Logan Adams Date: Thu, 9 Jan 2025 08:36:55 -0800 Subject: [PATCH 1/2] 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 From 2385831225ab64e85110545762bcb01002e5272a Mon Sep 17 00:00:00 2001 From: Logan Adams Date: Fri, 10 Jan 2025 13:36:30 -0800 Subject: [PATCH 2/2] Update pinned to latest official relase --- .github/workflows/nv-a6000.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/nv-a6000.yml b/.github/workflows/nv-a6000.yml index 0eee9d057d03..0547431e3099 100644 --- a/.github/workflows/nv-a6000.yml +++ b/.github/workflows/nv-a6000.yml @@ -42,7 +42,7 @@ jobs: run: | git clone https://github.com/huggingface/transformers cd transformers - git checkout v4.47.0 + git checkout v4.47.1 git rev-parse --short HEAD python -m pip install . - name: Install deepspeed