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
Add ONNX weights (+transformers.js support)
#2
by Xenova HF Staff - opened
Hi there! First of all, really great work with the model... I love seeing the MTEB leaderboard being switched up.
This PR adds ONNX weights for the model, both full-precision (fp32; model.onnx) as well as 8-bit precision (int8; model_quantized.onnx), so that it can be used with Transformers.js. Example usage can be found here.
Xenova changed pull request title from Add ONNX weights to Add ONNX weights (+transformers.js support)
I also added the transformers.js tag for discoverability.
It is great! Thanks for your contribution!
SeanLee97 changed pull request status to merged
Thanks! I’ll open up another PR so it shows both transformers and transformers.js library tags.