From ba56ed0869eb4bbeb1c04af7f62a04350150e8d4 Mon Sep 17 00:00:00 2001
From: MariaHei <44357394+MariaHei@users.noreply.github.com>
Date: Thu, 28 Mar 2024 21:26:27 +0000
Subject: [PATCH] Update installs in image classification doc (#29947)

Trainer with PyTorch now requires accelerate to be installed.

Partly resolves huggingface/transformers#29174
---
 docs/source/en/tasks/image_classification.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/docs/source/en/tasks/image_classification.md b/docs/source/en/tasks/image_classification.md
index 22a568f5e4..30c517f3be 100644
--- a/docs/source/en/tasks/image_classification.md
+++ b/docs/source/en/tasks/image_classification.md
@@ -43,7 +43,7 @@ The task illustrated in this tutorial is supported by the following model archit
 Before you begin, make sure you have all the necessary libraries installed:
 
 ```bash
-pip install transformers datasets evaluate
+pip install transformers datasets evaluate accelerate
 ```
 
 We encourage you to log in to your Hugging Face account to upload and share your model with the community. When prompted, enter your token to log in: