Test DeepSpeed in Fabric CI (#16458)
Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
This commit is contained in:
parent
9faa25f86f
commit
b30e948ba4
|
@ -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
|
||||
|
|
|
@ -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"
|
||||
|
|
Loading…
Reference in New Issue