Feature Extraction
sentence-transformers
ONNX
Safetensors
OpenVINO
Transformers
Transformers.js
English
bert
mteb
sentence_embedding
feature_extraction
Eval Results (legacy)
text-embeddings-inference
Instructions to use WhereIsAI/UAE-Large-V1 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- sentence-transformers
How to use WhereIsAI/UAE-Large-V1 with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("WhereIsAI/UAE-Large-V1") sentences = [ "The weather is lovely today.", "It's so sunny outside!", "He drove to the stadium." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [3, 3] - Transformers
How to use WhereIsAI/UAE-Large-V1 with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("feature-extraction", model="WhereIsAI/UAE-Large-V1")# Load model directly from transformers import AutoTokenizer, AutoModel tokenizer = AutoTokenizer.from_pretrained("WhereIsAI/UAE-Large-V1") model = AutoModel.from_pretrained("WhereIsAI/UAE-Large-V1") - Transformers.js
How to use WhereIsAI/UAE-Large-V1 with Transformers.js:
// npm i @huggingface/transformers import { pipeline } from '@huggingface/transformers'; // Allocate pipeline const pipe = await pipeline('feature-extraction', 'WhereIsAI/UAE-Large-V1'); - Notebooks
- Google Colab
- Kaggle
how to perform reranking?
#13
by erjieyong - opened
Hi, may i know if there's any cross-encoder type reranking? Appreciate clarification on how we can perform reranking. Are the below steps correct?
- use the retrieval Prompt.C encode
- compare similarities and retrieve k number of relevant docs
- Remove Prompt.C by setting 'prompt = None' and encode the k number of relevant docs found in step 2
- rerank the k number of relevant docs based on embeddings from step 3
Hi, it doesn't support cross-encoder now.
The steps are correct. Prompts.C is applied for the query, not documents.
Thanks @SeanLee97 . May i confirm that these steps is also how the reranking benchmarks are achieved?