refine the comment

huggingface · Oct 21, 2023 · 06a429a · 06a429a
1 parent f5d241e
commit 06a429a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/optimum/onnxruntime/trainer.py b/optimum/onnxruntime/trainer.py
@@ -476,7 +476,7 @@ def _inner_training_loop(
         if self.is_deepspeed_enabled:
             if is_deepspeed_zero3_enabled() and not ort_support_stage3:
                 raise NotImplementedError(
-                    "`ORTTrainer` does not support ZeRO stage 3 for the moment. Please use DeepSpeed stage 1 or 2 instead."
+                    "`ORTTrainer` does not support ZeRO stage 3 for the moment. Please use DeepSpeed stage 1 or 2 instead or consider to update ONNX Runtime to 1.17 or later."
                 )
             if args.bf16:
                 warnings.warn(