SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix typo in deepspeed documentation (#13482)

Browse Source 
* Fix typo in deepspeed documentation

* Add missing import in deepspeed configuration
This commit is contained in:
Aleksander Smywiński-Pohl
2021-09-08 20:24:10 +02:00
committed by GitHub
parent e1f6e4903a
commit c37573806a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docs/source/main_classes/deepspeed.rst
View File

@@ -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:
.. 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