Test DeepSpeed in Fabric CI (#16458)

Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
This commit is contained in:
Adrian Wälchli 2023-01-23 12:40:56 +01:00 committed by GitHub
parent 9faa25f86f
commit b30e948ba4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 1 deletions

View File

@ -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

View File

@ -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"