From 3d388a1aeada7d9b7b334ba915d684d3ac3c2066 Mon Sep 17 00:00:00 2001 From: Abhishek Jindal Date: Wed, 1 Feb 2023 12:00:43 -0800 Subject: [PATCH] change deepspeed version in warning from 0.7.3 to 0.8.0 (#14527) ### 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. --- orttraining/orttraining/python/training/optim/_ds_modifier.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/orttraining/orttraining/python/training/optim/_ds_modifier.py b/orttraining/orttraining/python/training/optim/_ds_modifier.py index dc2d4c091c..8a579f4ccd 100644 --- a/orttraining/orttraining/python/training/optim/_ds_modifier.py +++ b/orttraining/orttraining/python/training/optim/_ds_modifier.py @@ -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