Skip to content

Commit 18346f2

Browse files
committed
readme update
1 parent e982e80 commit 18346f2

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

README.md

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -67,6 +67,9 @@ pip install "rerankers[flashrank]"
6767
# Note: RankLLM is only supported on Python 3.10+! This will not work with Python 3.9
6868
pip install "rerankers[rankllm]"
6969

70+
# To support LLM-Layerwise rerankers (which need flash-attention installed)
71+
pip install "rerankers[llmlayerwise]"
72+
7073
# All of the above
7174
pip install "rerankers[all]"
7275
```
@@ -119,12 +122,12 @@ ranker = Reranker("rankllm", api_key = API_KEY)
119122
# RankLLM with specified GPT models
120123
ranker = Reranker('gpt-4-turbo', model_type="rankllm", api_key = API_KEY)
121124

122-
# EXPERIMENTAL: RankLLM with RankZephyr
123-
ranker = Reranker('rankzephyr')
124-
125125
# ColBERTv2 reranker
126126
ranker = Reranker("colbert")
127127

128+
# LLM Layerwise Reranker
129+
ranker = Reranker('llm-layerwise')
130+
128131
# ... Or a non-default colbert model:
129132
ranker = Reranker(model_name_or_path, model_type = "colbert")
130133

0 commit comments

Comments
 (0)