PrithivirajDamodaran / FlashRank

Lite & Super-fast re-ranking for your search & retrieval pipelines. Supports SoTA Listwise and Pairwise reranking based on LLMs and cross-encoders and more. Created by Prithivi Da, open for PRs & Collaborations.
Apache License 2.0
441 stars 37 forks source link

Support BAAI/bge-reranker-base? #23

Closed vipinap98 closed 3 weeks ago

vipinap98 commented 3 weeks ago

Wont it be good if we have support for BAAI/bge-reranker-base model? seems it works pretty well

PrithivirajDamodaran commented 3 weeks ago

No, flashrank philosophy is choose mini / tiny models. So for pointwise ranking with cross-encoder based models > 1 GB (like BAAI/bge-reranker-base) are doesnt make sense. For LLM listwise ranking using LLMs its a different criteria.