From b30e948ba41be4e50ba9669ecc8dcf6c02b39daa Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Adrian=20W=C3=A4lchli?= Date: Mon, 23 Jan 2023 12:40:56 +0100 Subject: [PATCH] Test DeepSpeed in Fabric CI (#16458) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-authored-by: Carlos Mocholí --- .github/workflows/ci-tests-fabric.yml | 1 + requirements/fabric/strategies.txt | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/.github/workflows/ci-tests-fabric.yml b/.github/workflows/ci-tests-fabric.yml index 0b5bb7408b..f715e1e20c 100644 --- a/.github/workflows/ci-tests-fabric.yml +++ b/.github/workflows/ci-tests-fabric.yml @@ -108,6 +108,7 @@ jobs: FREEZE_REQUIREMENTS: 1 run: | pip install -e .[test] "pytest-timeout" --upgrade --find-links ${TORCH_URL} + pip install -r requirements/fabric/strategies.txt pip list - name: Adjust tests diff --git a/requirements/fabric/strategies.txt b/requirements/fabric/strategies.txt index c5fbf3dc5b..2ca57854ea 100644 --- a/requirements/fabric/strategies.txt +++ b/requirements/fabric/strategies.txt @@ -1,3 +1,3 @@ # NOTE: the upper bound for the package version is only set for CI stability, and it is dropped while installing this package # in case you want to preserve/enforce restrictions on the latest compatible version, add "strict" as an in-line comment -deepspeed>=0.6.0, <=0.7.0 +deepspeed>=0.6.0, <=0.7.0; python_version>"3.7" and platform_system != "Windows"