[Doc] Fix Title of the Transformers GLUE example (#39605)

Signed-off-by: woshiyyya <xiaoyunxuan1998@gmail.com>
ray-project · Sep 13, 2023 · b00d029 · b00d029
1 parent 5cd72b9
commit b00d029
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/doc/source/train/examples.rst b/doc/source/train/examples.rst
@@ -45,8 +45,8 @@ Intermediate
     - `DreamBooth fine-tuning of Stable Diffusion with Ray Train <https://github.com/ray-project/ray/tree/master/doc/source/templates/05_dreambooth_finetuning>`_
   * - Lightning
     - :ref:`Model Training with PyTorch Lightning and Ray Data <lightning_advanced_example>`
-  * - Accelerate
-    - :ref:`Fine-tuning a Text Classifier on GLUE Benchmark with HF Accelerate. <train_transformers_accelerate_example>`
+  * - Transformers
+    - :ref:`Fine-tuning a Text Classifier on GLUE Benchmark with HF Transformers. <train_transformers_glue_example>`
 
 
 Advanced

diff --git a/doc/source/train/examples/transformers/huggingface_text_classification.ipynb b/doc/source/train/examples/transformers/huggingface_text_classification.ipynb
@@ -4,7 +4,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "(train_transformers_accelerate_example)=\n",
+    "(train_transformers_glue_example)=\n",
     "\n",
     "# Fine-tune a 🤗 Transformers model"
    ]

diff --git a/python/ray/train/examples/accelerate/accelerate_torch_trainer.py b/python/ray/train/examples/accelerate/accelerate_torch_trainer.py
@@ -19,6 +19,7 @@
     set_seed,
 )
 
+import ray
 import ray.train
 from ray.train import DataConfig, ScalingConfig, Checkpoint
 from ray.train.torch import TorchTrainer