Skip to content

Commit ed49eb1

Browse files
authored
update default cross-encoder model to Xenova/ms-marco-MiniLM-L-6-v2 across docs (#41)
1 parent bc1fe22 commit ed49eb1

4 files changed

Lines changed: 12 additions & 12 deletions

File tree

docs/reference.md

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -86,13 +86,13 @@ SEARCH docs SIMILAR TO 'hello' LIMIT 5
8686
### Cross-encoder reranking (RERANK default)
8787

8888
```
89-
cross-encoder/ms-marco-MiniLM-L-6-v2
89+
Xenova/ms-marco-MiniLM-L-6-v2
9090
```
9191

9292
| Model | Notes |
9393
|---|---|
94-
| `cross-encoder/ms-marco-MiniLM-L-6-v2` | Default. Fast passage reranker |
95-
| `cross-encoder/ms-marco-MiniLM-L-12-v2` | Larger, higher quality |
94+
| `Xenova/ms-marco-MiniLM-L-6-v2` | Default. Fast passage reranker |
95+
| `Xenova/ms-marco-MiniLM-L-12-v2` | Larger, higher quality |
9696
| `BAAI/bge-reranker-base` | Strong general-purpose reranker |
9797
| `BAAI/bge-reranker-large` | Highest quality, slower |
9898

docs/search.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -377,15 +377,15 @@ SEARCH articles SIMILAR TO 'attention mechanism in transformers' LIMIT 10 USING
377377
Custom cross-encoder model:
378378
```sql
379379
SEARCH articles SIMILAR TO 'semantic search' LIMIT 5
380-
RERANK MODEL 'cross-encoder/ms-marco-MiniLM-L-6-v2'
380+
RERANK MODEL 'Xenova/ms-marco-MiniLM-L-6-v2'
381381
```
382382

383-
**Default cross-encoder model:** `cross-encoder/ms-marco-MiniLM-L-6-v2`
383+
**Default cross-encoder model:** `Xenova/ms-marco-MiniLM-L-6-v2`
384384

385385
| Model | Notes |
386386
|---|---|
387-
| `cross-encoder/ms-marco-MiniLM-L-6-v2` | Default. Fast and accurate for passage reranking |
388-
| `cross-encoder/ms-marco-MiniLM-L-12-v2` | Larger, higher quality, slower |
387+
| `Xenova/ms-marco-MiniLM-L-6-v2` | Default. Fast and accurate for passage reranking |
388+
| `Xenova/ms-marco-MiniLM-L-12-v2` | Larger, higher quality, slower |
389389
| `BAAI/bge-reranker-base` | BGE reranker, strong general-purpose performance |
390390
| `BAAI/bge-reranker-large` | Highest quality BGE reranker, slower |
391391

resources/Features.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -431,7 +431,7 @@ SEARCH medical_records SIMILAR TO 'neuromuscular junction antibody acetylcholine
431431
AND sub_specialty IN ('neuromuscular', 'neuroimmunology')
432432
AND severity IN ('high', 'critical')
433433
AND year >= 2022
434-
RERANK MODEL 'cross-encoder/ms-marco-MiniLM-L-6-v2'
434+
RERANK MODEL 'Xenova/ms-marco-MiniLM-L-6-v2'
435435
```
436436

437437
---

tests/test_parser.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1003,10 +1003,10 @@ def test_rerank_flag_set(self):
10031003

10041004
def test_rerank_with_model(self):
10051005
node = parse(
1006-
"SEARCH col SIMILAR TO 'q' LIMIT 5 RERANK MODEL 'cross-encoder/ms-marco-MiniLM-L-6-v2'"
1006+
"SEARCH col SIMILAR TO 'q' LIMIT 5 RERANK MODEL 'Xenova/ms-marco-MiniLM-L-6-v2'"
10071007
)
10081008
assert node.rerank is True
1009-
assert node.rerank_model == "cross-encoder/ms-marco-MiniLM-L-6-v2"
1009+
assert node.rerank_model == "Xenova/ms-marco-MiniLM-L-6-v2"
10101010

10111011
def test_rerank_default_false(self):
10121012
node = parse("SEARCH col SIMILAR TO 'q' LIMIT 5")
@@ -1032,12 +1032,12 @@ def test_rerank_with_where(self):
10321032
def test_rerank_with_hybrid_where_and_model(self):
10331033
node = parse(
10341034
"SEARCH col SIMILAR TO 'q' LIMIT 5 USING HYBRID WHERE year > 2020 "
1035-
"RERANK MODEL 'cross-encoder/ms-marco-MiniLM-L-6-v2'"
1035+
"RERANK MODEL 'Xenova/ms-marco-MiniLM-L-6-v2'"
10361036
)
10371037
assert node.hybrid is True
10381038
assert node.query_filter is not None
10391039
assert node.rerank is True
1040-
assert node.rerank_model == "cross-encoder/ms-marco-MiniLM-L-6-v2"
1040+
assert node.rerank_model == "Xenova/ms-marco-MiniLM-L-6-v2"
10411041

10421042
def test_rerank_lowercase(self):
10431043
node = parse("SEARCH col SIMILAR TO 'q' LIMIT 5 rerank")

0 commit comments

Comments
 (0)