BAAI

/

bge-reranker-v2-m3

like 779

Follow

Beijing Academy of Artificial Intelligence 2.78k

Hello!

This pull request has been automatically generated from the Sentence Transformers backend-export Space.

Pull Request overview

• Add exported ONNX model model.onnx.

Tip:

Consider testing this pull request before merging by loading the model from this PR with the revision argument:

from sentence_transformers import CrossEncoder

# TODO: Fill in the PR number
pr_number = 2
model = CrossEncoder(
    "BAAI/bge-reranker-v2-m3",
    revision=f"refs/pr/{pr_number}",
    backend="onnx",
)

# Verify that everything works as expected
predictions = model.predict([
    ["Which planet is known as the Red Planet?", "Mars, known for its reddish appearance, is often referred to as the Red Planet."],
    ["Which planet is known as the Red Planet?", "Jupiter, the largest planet in our solar system, has a prominent red spot."],
])
print(predictions)