From 0ab3a417758c93e9f3005a69f363f9a1139a77cb Mon Sep 17 00:00:00 2001 From: Logan Adams Date: Fri, 7 Feb 2025 10:57:39 -0800 Subject: [PATCH 1/2] Update A6000 tests Signed-off-by: Logan Adams --- .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 5631d318d027..75a682cb0fc1 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.1 + #git checkout v4.47.1 git rev-parse --short HEAD python -m pip install . - name: Install deepspeed From b7e8118f58af7a416b199bd2360c7cdc0c3386c8 Mon Sep 17 00:00:00 2001 From: Logan Adams <114770087+loadams@users.noreply.github.com> Date: Fri, 7 Feb 2025 12:50:36 -0800 Subject: [PATCH 2/2] Update .github/workflows/nv-a6000.yml Co-authored-by: Stas Bekman Signed-off-by: Logan Adams --- .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 75a682cb0fc1..502b29b44c13 100644 --- a/.github/workflows/nv-a6000.yml +++ b/.github/workflows/nv-a6000.yml @@ -42,7 +42,8 @@ jobs: run: | git clone https://github.com/huggingface/transformers cd transformers - #git checkout v4.47.1 + # if you need to use an older transformers version temporarily in case of breakage + # git checkout v4.47.1 git rev-parse --short HEAD python -m pip install . - name: Install deepspeed