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

change deepspeed version in warning from 0.7.3 to 0.8.0 (#14527)

Browse source 
### Description
change deepspeed version in warning from 0.7.3 to 0.8.0



### Motivation and Context
The version was updated for Deepspeed support in ORT from 0.7.3 to 0.8.0
but wasn't updated in the warnings message and this PR is to fix that.
This commit is contained in:
Abhishek Jindal 2023-02-01 12:00:43 -08:00 committed by GitHub
parent b1a17188a6
commit 3d388a1aea
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -42,7 +42,7 @@ class DeepSpeedZeROModifier(FP16OptimizerModifier):
if ds_version > Version("0.8.0") or ds_version < Version("0.4.0"):
warnings.warn(
"Skip modifying optimizer because of unsupported DeepSpeed version {}, "
"supported version: 0.4.0 - 0.7.3.".format(deepspeed.__version__),
"supported version: 0.4.0 - 0.8.0.".format(deepspeed.__version__),
UserWarning,
)
return False