From 08c2e37e325cbba3e648cfd26e1c2547ed3b48fa Mon Sep 17 00:00:00 2001
From: Ella Charlaix <ella@huggingface.co>
Date: Wed, 29 Jan 2025 18:15:20 +0100
Subject: [PATCH] Update ort training installation instructions

---
 README.md | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/README.md b/README.md
index 9a81e69e12..4b5a8380c6 100644
--- a/README.md
+++ b/README.md
@@ -239,6 +239,13 @@ You can find more examples in the [documentation](https://huggingface.co/docs/op
 
 ### ONNX Runtime
 
+
+Before you begin, make sure you have all the necessary libraries installed :
+
+```bash
+pip install optimum[onnxruntime-training]
+```
+
 ```diff
 - from transformers import Trainer, TrainingArguments
 + from optimum.onnxruntime import ORTTrainer, ORTTrainingArguments