Skip to content

Commit

Permalink
Fix model name typo in example (#3028)
Browse files Browse the repository at this point in the history
The model in the example: `msmarco-distilroberta-base-v3` is missing from huggingface.

The existing ones are `msmarco-distilbert-base-v3` and `msmarco-roberta-base-v3`.

Changed the model name to the former one since it downloads faster.
  • Loading branch information
programmer-ke authored Oct 30, 2024
1 parent 210ea8b commit 68dfbe6
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/pretrained-models/msmarco-v3.md
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ The training data consists of over 500k examples, while the complete corpus con
```python
from sentence_transformers import SentenceTransformer, util

model = SentenceTransformer("msmarco-distilroberta-base-v3")
model = SentenceTransformer("msmarco-distilbert-base-v3")

query_embedding = model.encode("How big is London")
passage_embedding = model.encode("London has 9,787,426 inhabitants at the 2011 census")
Expand Down

0 comments on commit 68dfbe6

Please sign in to comment.