From 65a0b13deee1236f32027bed94811e0ee0115b64 Mon Sep 17 00:00:00 2001 From: Ashwini Khade Date: Mon, 8 Jan 2024 11:24:10 -0800 Subject: [PATCH] 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