Skip to content

Commit

Permalink
[docs] Fix doc format (#28684)
Browse files Browse the repository at this point in the history
* fix hfoptions

* revert changes to other files

* fix
  • Loading branch information
stevhliu authored Jan 24, 2024
1 parent 8278b15 commit f40b87d
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion docs/source/en/deepspeed.md
Original file line number Diff line number Diff line change
Expand Up @@ -956,6 +956,8 @@ engine = deepspeed.initialize(model=model, config_params=ds_config, ...)
</hfoption>
<hfoption id="non-pretrained model">

[`HfDeepSpeedConfig`] is not required for ZeRO-1 or ZeRO-2.

```py
from transformers.integrations import HfDeepSpeedConfig
from transformers import AutoModel, AutoConfig
Expand All @@ -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.
</hfoption>
</hfoptions>

### Non-Trainer ZeRO Inference

Expand Down

0 comments on commit f40b87d

Please sign in to comment.