Update ort training installation instructions (#2173)

huggingface · Jan 30, 2025 · 90c4146 · 90c4146
1 parent b755036
commit 90c4146
Showing 1 changed file with 7 additions and 0 deletions.
diff --git 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