Skip to content

Commit 6525bcf

Browse files
committed
update reranker
1 parent 413000e commit 6525bcf

2 files changed

Lines changed: 11 additions & 0 deletions

File tree

  • FlagEmbedding/inference/reranker/decoder_only
  • examples/evaluation

FlagEmbedding/inference/reranker/decoder_only/base.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -165,9 +165,12 @@ def __init__(
165165
query_max_length=query_max_length,
166166
max_length=max_length,
167167
normalize=normalize,
168+
prompt=prompt,
168169
**kwargs
169170
)
170171

172+
self.prompt = prompt
173+
171174
self.tokenizer = AutoTokenizer.from_pretrained(
172175
model_name_or_path,
173176
cache_dir=cache_dir,

examples/evaluation/README.md

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -84,6 +84,14 @@ In this section, we will first introduce the commonly used arguments across all
8484

8585
## Usage
8686

87+
### Requirements
88+
89+
You need install faiss for evaluation:
90+
91+
```shell
92+
pip install https://github.com/kyamagu/faiss-wheels/releases/download/v1.7.3/faiss_gpu-1.7.3-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
93+
```
94+
8795
### 1. MTEB
8896

8997
For MTEB, we primarily use the official [MTEB](https://github.com/embeddings-benchmark/mteb) code, which only supports the assessment of embedders. Moreover, it restricts the output format of the evaluation results to JSON. We have introduced the following new arguments:

0 commit comments

Comments
 (0)