Text Ranking
sentence-transformers
Safetensors
Transformers
multilingual
t5gemma2
text2text-generation
reranker
encoder-decoder
FBNL
Retrieval
RAG
Instructions to use KaLM-Embedding/KaLM-Reranker-V1-Large with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use KaLM-Embedding/KaLM-Reranker-V1-Large with sentence-transformers:
from sentence_transformers import CrossEncoder model = CrossEncoder("KaLM-Embedding/KaLM-Reranker-V1-Large") query = "Which planet is known as the Red Planet?" passages = [ "Venus is often called Earth's twin because of its similar size and proximity.", "Mars, known for its reddish appearance, is often referred to as the Red Planet.", "Jupiter, the largest planet in our solar system, has a prominent red spot.", "Saturn, famous for its rings, is sometimes mistaken for the Red Planet." ] scores = model.predict([(query, passage) for passage in passages]) print(scores) - Transformers
How to use KaLM-Embedding/KaLM-Reranker-V1-Large with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoProcessor, AutoModelForMultimodalLM processor = AutoProcessor.from_pretrained("KaLM-Embedding/KaLM-Reranker-V1-Large") model = AutoModelForMultimodalLM.from_pretrained("KaLM-Embedding/KaLM-Reranker-V1-Large", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Commit History
30 days ago
-
Update README.md
Verified signatureec1a5b1
-
Update README.md
Verified signature70b68ae -
Update vllm_support/README.md
Verified signature7a1ef3b -
Update vLLM support and remove build artifacts
Verified signature7f339dc -
Update vllm_support/src/kalm_t5gemma2_vllm_plugin/constants.py
Verified signature8b567a7 -
Update README.md
Verified signature63a7eb4
-
Update README.md
Verified signature7a48c18
-
Update README.md
Verified signature8acd73d -
Update README.md
Verified signaturee844afe -
Add sentence-transformers CrossEncoder support (#2)
- Add sentence-transformers CrossEncoder support (566db9ad7318cdd2648cca1b8e4c64e909a8357e)730398d
-
fix(reranker): avoid re-computing the first batch in predict()'s batch-size probe to reduce additional computational effort (#1)
- fix(reranker): avoid re-computing the first batch in predict()'s batch-size probe to reduce additional computational effort (621b3be26e5b4585c31d770e3def0c34f7e13fbf)e4e0ad3
-
Update README.md
Verified signaturebc5cb8f -
Upload 34 files
Verified signature6f7a484
-
Update README.md
Verified signaturebdc942d
-
Update README.md
Verified signature778c96c -
Update README.md
Verified signature147120c -
Update README.md
Verified signaturea7b5a44 -
Update README.md
Verified signatured30dddb -
Update kalm_reranker.py
Verified signature230e162
-
Update README.md
Verified signature9c69698 -
Update README.md
Verified signature673fcd0 -
add modeling file
Verified signature59102d6 -
Update README.md
Verified signature5ee1357 -
Create README.md
Verified signature9bb9732 -
Upload 6 files
Verified signature42440c6 -
Upload model.safetensors
Verified signature5536346 -
Upload 4 files
Verified signaturec8d34de
-
initial commit
Verified signature08d0c1b