saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-25 22:26:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

update deepspeed version 0.8.3 (#15415)

Browse source 
### Description
<!-- Describe your changes. -->
Update the support deepspeed to 0.8.3 as it's the latest version


### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here. -->
This will fix the error of `Skip modifying optimizer because of
unsupported DeepSpeed version`

Co-authored-by: ruiren <ruiren@microsoft.com>
This commit is contained in:
Rui Ren 2023-04-07 17:59:50 -07:00 committed by GitHub
parent 666aff56a4
commit 5e2f46df2b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
orttraining/orttraining/python/training/optim/_ds_modifier.py
View file

@ -39,10 +39,10 @@ class DeepSpeedZeROModifier(FP16OptimizerModifier):
# it's safe to update the version supporting list. Otherwise, or the file is moved or renamed,
# we need to check the implementation of these functions in detail.
ds_version = Version(deepspeed.__version__)
if ds_version > Version("0.8.0") or ds_version < Version("0.4.0"):
if ds_version > Version("0.8.3") or ds_version < Version("0.4.0"):
warnings.warn(
"Skip modifying optimizer because of unsupported DeepSpeed version {}, "
"supported version: 0.4.0 - 0.8.0.".format(deepspeed.__version__),
"supported version: 0.4.0 - 0.8.3.".format(deepspeed.__version__),
UserWarning,
)
return False