Skip to content

Commit aa786c3

Browse files
authored
Fix tag train documentation (#269)
* Update README.md fix tag for training installation * Update README.md fix tag for pip install training mode
1 parent e0fe9d1 commit aa786c3

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

‎README.md‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,7 @@ m2v_model.save_pretrained("m2v_model")
9292
After distillation, you can also fine-tune your own classification models on top of the distilled model, or on a pre-trained model. First, make sure you install the `training` extras with:
9393

9494
```bash
95-
pip install model2vec[training]
95+
pip install model2vec[train]
9696
```
9797

9898
Then, you can fine-tune a model as follows:

‎model2vec/train/README.md‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ We support both single and multi-label classification, which work seamlessly bas
99
To train, make sure you install the training extra:
1010

1111
```
12-
pip install model2vec[training]
12+
pip install model2vec[train]
1313
```
1414

1515
# Quickstart

0 commit comments

Comments
 (0)