Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

add support bge-reranker-v2.5-gemma-lightweight #368

Open
1 of 2 tasks
ziozzang opened this issue Aug 6, 2024 · 2 comments
Open
1 of 2 tasks

add support bge-reranker-v2.5-gemma-lightweight #368

ziozzang opened this issue Aug 6, 2024 · 2 comments

Comments

@ziozzang
Copy link

ziozzang commented Aug 6, 2024

Model description

bge-reranker-v2.5-gemma-lightweight 's performance is better bge-m3 :)

Please support model.

Open source status

  • The model implementation is available
  • The model weights are available

Provide useful links for the implementation

https://huggingface.co/BAAI/bge-reranker-v2.5-gemma2-lightweight

@RocketRider
Copy link

RocketRider commented Aug 22, 2024

Did just run into this issue as well. Would be great if we could support this new model.

Also the embedding model:
https://huggingface.co/BAAI/bge-multilingual-gemma2

@colexbruhn
Copy link

Chiming in to express interest as well. This model is great!

It might require adding a few extra runtime args for layer cutoff, compression, etc.
Also maybe the ability to choose prompts at inference time?

You can use bge-reranker-v2.5-gemma2-lightweight with the following different prompts:

  • Predict whether passage B contains an answer to query A.
  • Predict whether passages A and B have the same meaning.
  • Predict whether queries A and B are asking the same thing.
  • Predict whether argument A and counterargument B express contradictory opinions.

Really cool stuff, hoping this can get picked up!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants