Skip to content

feat: support bi-encoder models in TransformerSimilarityRanker #8245

@Amnah199

Description

@Amnah199

Is your feature request related to a problem? Please describe.
The current implementation of TransformerSimilarityRanker only supports cross-encoder models, which limits the use of bi-encoder models like ColBERT v2.0.

Related discussion: Colbert as reranker

Describe the solution you'd like
Update the TransformerSimilarityRanker or creating a new component to support bi-encoder models like ColBERT.

Describe alternatives you've considered
Leave the current implementation as is.

Additional context
Similar implementation: Llama ColbertRerank

Metadata

Metadata

Assignees

No one assigned

    Labels

    Contributions wanted!Looking for external contributionsP3Low priority, leave it in the backlogtype:featureNew feature or request

    Type

    No type

    Projects

    Status

    Todo

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions