Views
No views yet
python -m pip install optimum1from optimum.onnxruntime.modeling_ort import ORTModelForCustomTasks
2
3model = ORTModelForCustomTasks.from_pretrained("vamsibanda/sbert-all-roberta-large-v1-with-pooler")
4tokenizer = AutoTokenizer.from_pretrained("vamsibanda/sbert-all-roberta-large-v1-with-pooler")
5inputs = tokenizer("I love burritos!", return_tensors="pt")
6pred = model(**inputs)
7embedding = pred['pooler_output']