Downloads · 30 days
9
27% of all-time downloads
gte-small-dense
zeroshot/gte-small-dense
gte-small-dense is a feature extraction model from zeroshot. 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 gte-small embeddings model created with the DeepSparse Optimum integration.
All-time downloads
33
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
- Oct 12, 2023
- Updated
- Oct 12, 2023
- SHA
- 3176d6b4
Model card
From the Hugging Face model README
This is the ONNX variant of the gte-small embeddings model created with the DeepSparse Optimum integration.
To replicate 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 = "thenlper/gte-small"
# 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("gte-small-dense")
model.save_pretrained(onnx_path)
tokenizer.save_pretrained(onnx_path)
- Task
- Feature Extraction
- Library
- transformers
- Type
- bert
- License
- mit
- Languages
- en
- Created
- Oct 12, 2023
- Updated
- Oct 12, 2023