diff --git a/docs/source/en/deepspeed.md b/docs/source/en/deepspeed.md index f9b7fef9520b..77c198bc56d7 100644 --- a/docs/source/en/deepspeed.md +++ b/docs/source/en/deepspeed.md @@ -956,6 +956,8 @@ engine = deepspeed.initialize(model=model, config_params=ds_config, ...) +[`HfDeepSpeedConfig`] is not required for ZeRO-1 or ZeRO-2. + ```py from transformers.integrations import HfDeepSpeedConfig from transformers import AutoModel, AutoConfig @@ -969,7 +971,8 @@ model = AutoModel.from_config(config) engine = deepspeed.initialize(model=model, config_params=ds_config, ...) ``` -[`HfDeepSpeedConfig`] is not required for ZeRO-1 or ZeRO-2. + + ### Non-Trainer ZeRO Inference