gte-base-onnx / README.md
kais-radwan's picture
Upload README.md
03be7a2 verified
|
raw
history blame
418 Bytes

gte-base

This is a fork from https://huggingface.co/thenlper/gte-base that was optimized using Optimum and ONNX for better performance.

Usage

from transformers import AutoTokenizer
from optimum.onnxruntime import ORTModelForFeatureExtraction

tokenizer = AutoTokenizer.from_pretrained("kais-radwan/gte-base-onnx")
model = ORTModelForFeatureExtraction.from_pretrained("kais-radwan/gte-base-onnx")