From c37573806ab3526dd805c49cbe2489ad4d68a9d7 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Aleksander=20Smywi=C5=84ski-Pohl?= Date: Wed, 8 Sep 2021 20:24:10 +0200 Subject: [PATCH] Fix typo in deepspeed documentation (#13482) * Fix typo in deepspeed documentation * Add missing import in deepspeed configuration --- docs/source/main_classes/deepspeed.rst | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/source/main_classes/deepspeed.rst b/docs/source/main_classes/deepspeed.rst index 18463bd3f1..db639bb53d 100644 --- a/docs/source/main_classes/deepspeed.rst +++ b/docs/source/main_classes/deepspeed.rst @@ -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