Fix typo in deepspeed documentation (#13482)
* Fix typo in deepspeed documentation * Add missing import in deepspeed configuration
This commit is contained in:
parent
e1f6e4903a
commit
c37573806a
|
@ -1728,7 +1728,7 @@ For example for a pretrained model:
|
|||
.. code-block:: python
|
||||
|
||||
from transformers.deepspeed import HfDeepSpeedConfig
|
||||
from transformers import AugoModel
|
||||
from transformers import AutoModel, deepspeed
|
||||
|
||||
ds_config = { ... } # deepspeed config object or path to the file
|
||||
# must run before instantiating the model
|
||||
|
@ -1741,7 +1741,7 @@ or for non-pretrained model:
|
|||
.. code-block:: python
|
||||
|
||||
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
|
||||
# must run before instantiating the model
|
||||
|
|
Loading…
Reference in New Issue