From 52f4681535a9a5dac46b32022b9c4fa9ff1b7c09 Mon Sep 17 00:00:00 2001 From: Ashwini Khade Date: Mon, 18 Dec 2023 13:16:14 -0800 Subject: [PATCH] install accelerate package --- .../ortmodule/stage1/requirements_torch_nightly/requirements.txt | 1 + 1 file changed, 1 insertion(+) 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 fc8e542cb9833..0cd5e5c5d5c46 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,4 +1,5 @@ scikit-learn packaging==21.3 transformers==v4.30.0 +accelerate==0.20.1 wget