Skip to content

Commit b0f1d73

Browse files
committed
update reranker v2
1 parent 282776f commit b0f1d73

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

FlagEmbedding/llm_reranker/README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ print(scores) # [0.00027803096387751553, 0.9948403768236574]
6666
```python
6767
from FlagEmbedding import FlagLLMReranker
6868
reranker = FlagLLMReranker('BAAI/bge-reranker-v2-gemma', use_fp16=True) # Setting use_fp16 to True speeds up computation with a slight performance degradation
69-
# reranker = FlagLLMReranker('BAAI/bge-reranker-v2-gemma', use_fp16=True) # You can also set use_bf16=True to speed up computation with a slight performance degradation
69+
# reranker = FlagLLMReranker('BAAI/bge-reranker-v2-gemma', use_bf16=True) # You can also set use_bf16=True to speed up computation with a slight performance degradation
7070

7171
score = reranker.compute_score(['query', 'passage'])
7272
print(score)

0 commit comments

Comments
 (0)