Downloads · 30 days
14
5% of all-time downloads
bge-small-en-v1.5-dense
RedHatAI/bge-small-en-v1.5-dense
bge-small-en-v1.5-dense is a feature extraction model from RedHatAI. Use it when you need embeddings to search or compare text. It is set up for transformers. The card lists the license as mit.
This is the ONNX variant of the bge-small-en-v1.5 embeddings model created with the DeepSparse Optimum integration.
All-time downloads
310
Public
Repo size
133 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.onnx133 MB · 99%
At a glance
- Task
- Feature Extraction
- Library
- transformers
- License
- mit
- Model type
- bert
- Access
- Public
- Created
- Sep 21, 2023
- Updated
- Nov 13, 2023
- SHA
- 1e268156
Model card
From the Hugging Face model README
This is the ONNX variant of the bge-small-en-v1.5 embeddings model created with the DeepSparse Optimum integration.
For ONNX export, run:
pip install git+https://github.com/neuralmagic/optimum-deepsparse.git
from optimum.deepsparse import DeepSparseModelForFeatureExtraction
from transformers.onnx.utils import get_preprocessor
from pathlib import Path
model_id = "BAAI/bge-small-en-v1.5"
# load model and convert to onnx
model = DeepSparseModelForFeatureExtraction.from_pretrained(model_id, export=True)
tokenizer = get_preprocessor(model_id)
# save onnx checkpoint and tokenizer
onnx_path = Path("bge-small-en-v1.5-dense")
model.save_pretrained(onnx_path)
tokenizer.save_pretrained(onnx_path)
- Task
- Feature Extraction
- Library
- transformers
- Type
- bert
- License
- mit
- Languages
- en
- Created
- Sep 21, 2023
- Updated
- Nov 13, 2023