Unverified Commit c3757380 authored by Aleksander Smywiński-Pohl's avatar Aleksander Smywiński-Pohl Committed by GitHub
Browse files

Fix typo in deepspeed documentation (#13482)

* Fix typo in deepspeed documentation

* Add missing import in deepspeed configuration
parent e1f6e490
...@@ -1728,7 +1728,7 @@ For example for a pretrained model: ...@@ -1728,7 +1728,7 @@ For example for a pretrained model:
.. code-block:: python .. code-block:: python
from transformers.deepspeed import HfDeepSpeedConfig from transformers.deepspeed import HfDeepSpeedConfig
from transformers import AugoModel from transformers import AutoModel, deepspeed
ds_config = { ... } # deepspeed config object or path to the file ds_config = { ... } # deepspeed config object or path to the file
# must run before instantiating the model # must run before instantiating the model
...@@ -1741,7 +1741,7 @@ or for non-pretrained model: ...@@ -1741,7 +1741,7 @@ or for non-pretrained model:
.. code-block:: python .. code-block:: python
from transformers.deepspeed import HfDeepSpeedConfig from transformers.deepspeed import HfDeepSpeedConfig
from transformers import AugoModel, AutoConfig from transformers import AutoModel, AutoConfig, deepspeed
ds_config = { ... } # deepspeed config object or path to the file ds_config = { ... } # deepspeed config object or path to the file
# must run before instantiating the model # must run before instantiating the model
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment