From 6224c11b6cb52faeb5e99c6c31095e9cb7c47fa4 Mon Sep 17 00:00:00 2001 From: Ashwini Khade Date: Mon, 8 Jan 2024 10:14:02 -0800 Subject: [PATCH 1/3] Update transformer version --- .../stage1/requirements_torch_nightly/requirements.txt | 2 +- .../docker/scripts/training/ortmodule/stage2/requirements.txt | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch_nightly/requirements.txt b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch_nightly/requirements.txt index 0cd5e5c5d5c46..257005c5ab2fc 100644 --- a/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch_nightly/requirements.txt +++ b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch_nightly/requirements.txt @@ -1,5 +1,5 @@ scikit-learn packaging==21.3 -transformers==v4.30.0 +transformers==v4.36.0 accelerate==0.20.1 wget diff --git a/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage2/requirements.txt b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage2/requirements.txt index b4b265f65b69f..483b94e3f3ae8 100644 --- a/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage2/requirements.txt +++ b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage2/requirements.txt @@ -2,7 +2,7 @@ pandas scikit-learn numpy==1.21.6 ; python_version < '3.11' numpy==1.24.2 ; python_version >= '3.11' -transformers==v4.30.0 +transformers==v4.36.0 accelerate rsa==4.9 tensorboard==2.13.0 From 65a0b13deee1236f32027bed94811e0ee0115b64 Mon Sep 17 00:00:00 2001 From: Ashwini Khade Date: Mon, 8 Jan 2024 11:24:10 -0800 Subject: [PATCH 2/3] remove version hard coding for accelerate package --- .../stage1/requirements_torch_nightly/requirements.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch_nightly/requirements.txt b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch_nightly/requirements.txt index 257005c5ab2fc..f11d59ddd8089 100644 --- a/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch_nightly/requirements.txt +++ b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch_nightly/requirements.txt @@ -1,5 +1,5 @@ scikit-learn packaging==21.3 transformers==v4.36.0 -accelerate==0.20.1 +accelerate wget From aed682621b09605a4c72608a38ac7982806043af Mon Sep 17 00:00:00 2001 From: Ashwini Khade Date: Mon, 8 Jan 2024 16:12:09 -0800 Subject: [PATCH 3/3] Pin accelerate to working version --- .../stage1/requirements_torch_nightly/requirements.txt | 2 +- .../docker/scripts/training/ortmodule/stage2/requirements.txt | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch_nightly/requirements.txt b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch_nightly/requirements.txt index f11d59ddd8089..01fa7b0ff956e 100644 --- a/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch_nightly/requirements.txt +++ b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage1/requirements_torch_nightly/requirements.txt @@ -1,5 +1,5 @@ scikit-learn packaging==21.3 transformers==v4.36.0 -accelerate +accelerate==0.25.0 wget diff --git a/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage2/requirements.txt b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage2/requirements.txt index 483b94e3f3ae8..2b557f2aee00f 100644 --- a/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage2/requirements.txt +++ b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage2/requirements.txt @@ -3,7 +3,7 @@ scikit-learn numpy==1.21.6 ; python_version < '3.11' numpy==1.24.2 ; python_version >= '3.11' transformers==v4.36.0 -accelerate +accelerate==0.25.0 rsa==4.9 tensorboard==2.13.0 h5py