Skip to content

Commit 90c4146

Browse files
authored
Update ort training installation instructions (#2173)
1 parent b755036 commit 90c4146

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

README.md

+7
Original file line numberDiff line numberDiff line change
@@ -239,6 +239,13 @@ You can find more examples in the [documentation](https://huggingface.co/docs/op
239239

240240
### ONNX Runtime
241241

242+
243+
Before you begin, make sure you have all the necessary libraries installed :
244+
245+
```bash
246+
pip install optimum[onnxruntime-training]
247+
```
248+
242249
```diff
243250
- from transformers import Trainer, TrainingArguments
244251
+ from optimum.onnxruntime import ORTTrainer, ORTTrainingArguments

0 commit comments

Comments
 (0)